Hi, I'm not sure why I'm failing on test 1. Any ideas?

import sys

def main():
    ans = []
    for line in sys.stdin:
        nums = line.split()
        ans.extend([int(num) ** 0.5 for num in nums if num])
    for root in ans[::-1]:
        print("%.4f" % root)

Hey, you wrote your code in function. Just paste your code out of function