I’d like to add typehints to a function that accepts either np.float32 arrays or np.float64 arrays and returns the same type:

I am having trouble getting the numpy.typing.mypy_plugin to work.
Following the example of how to use NBitBase, I get different output from mypy.

When I run mypy main.py --strict on