from numpy.typing import NDArray  # should work