TST: Fix test in common.labels (#747)

* Fix assertion in tests/test_common/test_labels.py -- we now get back pandas.arrays.StringArray * Change dependency to 'dask[dataframe]' to squelch warning
vocalpy · May 1, 2024 · e2c1a71 · e2c1a71
1 parent 55d1a18
commit e2c1a71
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -25,7 +25,7 @@ license = {file = "LICENSE"}
 dependencies = [
     "attrs >=19.3.0",
     "crowsetta >=5.0.1",
-    "dask >=2.10.1",
+    "dask[dataframe] >=2.10.1",
     "evfuncs >=0.3.4",
     "joblib >=0.14.1",
     "pytorch-lightning >=2.0.7",

diff --git a/tests/test_common/test_labels.py b/tests/test_common/test_labels.py
@@ -1,7 +1,7 @@
 import copy
-import pathlib
 
 import numpy as np
+import pandas as pd
 import pytest
 
 import vak.common.files.spect
@@ -72,7 +72,7 @@ def test_from_df(config_type, model_name, audio_format, spect_format, annot_form
 
     out = vak.common.labels.from_df(df, dataset_path)
     assert isinstance(out, list)
-    assert all([isinstance(labels, np.ndarray) for labels in out])
+    assert all([isinstance(labels, (np.ndarray, pd.arrays.StringArray)) for labels in out])
 
 
 INTS_LABELMAP = {str(val): val for val in range(1, 20)}