Update src/spyglass/spikesorting/v1/metric_curation.py

Co-authored-by: Chris Brozdowski <[email protected]>
LorenFrankLab · Dec 1, 2023 · df2bfc8 · df2bfc8
1 parent fb31546
commit df2bfc8
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/spyglass/spikesorting/v1/metric_curation.py b/src/spyglass/spikesorting/v1/metric_curation.py
@@ -346,9 +346,7 @@ def get_labels(cls, key: dict):
         ) as io:
             nwbf = io.read()
             units = nwbf.objects[object_id].to_dataframe()
-        labels = dict(zip(units.index, units["curation_label"]))
-
-        return labels
+        return dict(zip(units.index, units["curation_label"]))
 
     @classmethod
     def get_merge_groups(cls, key: dict):