use set instead of np.unique

catalystneuro · Dec 19, 2024 · 366bc8f · 366bc8f
1 parent a58436a
commit 366bc8f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/neuroconv/tools/spikeinterface/spikeinterface.py b/src/neuroconv/tools/spikeinterface/spikeinterface.py
@@ -269,8 +269,8 @@ def _get_group_name(recording: BaseRecording) -> np.ndarray:
 
     # Validate group names against groups
     if groups is not None:
-        unique_groups = np.unique(groups)
-        unique_names = np.unique(group_names)
+        unique_groups = set(groups)
+        unique_names = set(group_names)
 
         if len(unique_names) != len(unique_groups):
             raise ValueError("The number of group names must match the number of groups")