input: Write a function to modify the given Python code to raise an exception if the input value is not an integer. If the input is an integer, the function should return the square root of the input.
  
  def square_root(x):
      if not isinstance(x, int):
          raise TypeError('Input must be an integer')
      return x**0.5
  
  assert square_root(4)==2.0
  assert square_root(9)==3.0
  try:
      square_root('four')
      raise AssertionError
  except TypeError:
      pass
  try:
      square_root(4.5)
      raise AssertionError
  except TypeError:
      pass
output: def square_root(x):
      if not isinstance(x, int):
          raise TypeError('Input must be an integer')
      return x**0.5