how to return square root in python code example

Example 1: python square root

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

Example 2: python get square root

import math

toSquare = 300
squared = math.sqrt(toSquare)