Problema a la hora de ejecutar

este es mi codigo y cuando trato de ejecutr sale un error de tiempo al ejecutar def count_digits(n: int) -> int: # Tu código aquí

count = 0

if n == 0:
    return 1

while n != 0:
    n = n // 10
    count += 1

return count