input: from typing import Union
  
  
  def square_root(x: Union[int, float]) -> float:
      """ Compute the square root of the given number.
      Raise TypeError if the input is not an integer.
      >>> square_root(4)
      2.0
      >>> square_root(9.0)
      Traceback (most recent call last):
          ...
      TypeError: Input must be an integer
      """
      if not isinstance(x, int):
          raise TypeError('Input must be an integer')
      return x**0.5
output:     if not isinstance(x, (int, float)):
          raise TypeError('Input must be an integer or float')
      return x**0.5