vocalpy / vak

A neural network framework for researchers studying acoustic communication
https://vak.readthedocs.io
BSD 3-Clause "New" or "Revised" License
78 stars 16 forks source link

TST: Fix test in `common.labels` #747

Closed NickleDave closed 6 months ago

NickleDave commented 6 months ago

This fixes a test that fails because in some cases pandas is converting the labels column to a pandas.arrays.StringArray.

I am fixing by just broadening the assertion to check whether isinstance(labels, (numpy.ndarray, pandas.arrays.StringArray)).

I need to investigate further but it doesn't seem to be breaking anything else right now.

I also added the dependency on the dask extra dataframe to squelch a warning we get every time we call dask.bag and dask.delayed.