Apply suggestions from code review

Co-authored-by: Chris Brozdowski <[email protected]>
LorenFrankLab · Jan 10, 2024 · 4ec0f50 · 4ec0f50
1 parent e4ae6d9
commit 4ec0f50
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/spyglass/decoding/v1/sorted_spikes.py b/src/spyglass/decoding/v1/sorted_spikes.py
@@ -321,10 +321,10 @@ def load_linear_position_info(key):
         position = np.asarray(position_df[["position_x", "position_y"]])
 
         linear_position_df = get_linearized_position(
-            position,
-            environment.track_graph,
-            environment.edge_order,
-            environment.edge_spacing,
+            position = position,
+            track_graph = environment.track_graph,
+            edge_order = environment.edge_order,
+            edge_spacing = environment.edge_spacing,
         )
 
         linear_position_df.insert(4, "speed", np.asarray(position_df.speed))

diff --git a/src/spyglass/spikesorting/imported.py b/src/spyglass/spikesorting/imported.py
@@ -31,8 +31,8 @@ def make(self, key):
 
     @classmethod
     def get_recording(cls, key):
-        raise NotImplementedError
+        raise NotImplementedError, "Imported spike sorting does not have a `get_recording` method"
 
     @classmethod
     def get_sorting(cls, key):
-        raise NotImplementedError
+        raise NotImplementedError, "Imported spike sorting does not have a `get_sorting` method"