update check for np array size

LorenFrankLab · Feb 12, 2024 · a4fcffa · a4fcffa
1 parent 6569210
commit a4fcffa
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/spyglass/spikesorting/analysis/v1/group.py b/src/spyglass/spikesorting/analysis/v1/group.py
@@ -97,7 +97,8 @@ def filter_units(
             if isinstance(unit_labels, str):
                 unit_labels = [unit_labels]
             if (
-                include_labels and np.all(~np.isin(unit_labels, include_labels))
+                include_labels.size > 0
+                and np.all(~np.isin(unit_labels, include_labels))
             ) or np.any(np.isin(unit_labels, exclude_labels)):
                 # if the unit does not have any of the include labels
                 # or has any of the exclude labels, skip