how to declare square root in python code example

Example: python square root

import math
a = input("what do you wnat to square root")
print(math.sqrt(a))