From 4ec0f50347b0a1f1ba95f9cf4fee38b17cb10148 Mon Sep 17 00:00:00 2001 From: Eric Denovellis Date: Wed, 10 Jan 2024 09:17:09 -0800 Subject: [PATCH] Apply suggestions from code review Co-authored-by: Chris Brozdowski --- src/spyglass/decoding/v1/sorted_spikes.py | 8 ++++---- src/spyglass/spikesorting/imported.py | 4 ++-- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/src/spyglass/decoding/v1/sorted_spikes.py b/src/spyglass/decoding/v1/sorted_spikes.py index 39ab52e3c..615453ad8 100644 --- 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 index bc9ce1b64..8c5f6cf72 100644 --- 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"