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

Co-authored-by: Chris Brozdowski <[email protected]>
khl02007 · Dec 8, 2023 · 80ef44d · 80ef44d
1 parent 7bfca04
commit 80ef44d
Showing 1 changed file with 4 additions and 9 deletions.
diff --git a/src/spyglass/spikesorting/v1/curation.py b/src/spyglass/spikesorting/v1/curation.py
@@ -449,15 +449,10 @@ def _list_to_merge_dict(
 
 
 def _reverse_associations(assoc_dict):
-    result = []
-
-    for key, values in assoc_dict.items():
-        if values:
-            result.append([key] + values)
-        else:
-            result.append([key])
-
-    return result
+        return [
+            [key] + values if values else [key]
+            for key, values in assoc_dict.items()
+        ]
 
 
 def _merge_dict_to_list(merge_groups: dict) -> List: