Faltou importar a biblioteca, eu estou usando o math.sqrt.
Então, se você está trabalhando com arrays, precisa usar o sqrt do módulo numpy e não do math.
Faltou importar a biblioteca, eu estou usando o math.sqrt.
Então, se você está trabalhando com arrays, precisa usar o sqrt do módulo numpy e não do math.