Can we specify dtypes of dask array as in numpy for typing?

Hey,

Is there any equivalent of numpy.typing.NDArray for Dask Arrays?

This would be of a great help to specify dtypes during typing:

def f(array: da.Array[np.float32]):
    ....

Thanks for your help!

Hi @louislt, welcome to Dask Discrouse forum!

Unfortunately, I don’t think this is possible, but I might be wrong. cc @fjetter @crusaderky.

1 Like