Fix: pyright error: Type "floating[Any]" is not assignable to return type "ndarray[Unknown, Unknown]"

Fix: pyright error: Type "floating[Any]" is not assignable to return type "ndarray[Unknown, Unknown]" #387