input: Write a function to modify the input Python code so that it raises an exception if the input value is not an integer. The function should return the square root of the input value if it is an integer.
  
  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('a')
      assert False
  except TypeError:
      assert True
output: def square_root(x):
      if not isinstance(x, int):
          raise TypeError('Input must be an integer')
      return x**0.5