A couple tiny changes for bug fixes. Should have tested previous comm…

…it. All is running successfully now.
catalystneuro · Dec 20, 2023 · 92d5b72 · 92d5b72
1 parent e1f35b7
commit 92d5b72
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -21,7 +21,6 @@ extend-exclude = '''
 '''
 
 [tool.isort]
-line-length = 79
 profile = "black"
 reverse_relative = true
 known_first_party = ["jazayeri_lab_to_nwb"]
diff --git a/src/jazayeri_lab_to_nwb/watters/nwb_converter.py b/src/jazayeri_lab_to_nwb/watters/nwb_converter.py
@@ -99,11 +99,11 @@ def temporally_align_data_interfaces(self):
                 lf_interface.set_aligned_timestamps(aligned_timestamps)
 
             # If sorting exists, register recording to it
-            if f"Sorting{probe_name}" in self.data_interface_classes:
+            if f"Sorting{probe_name}" in self.data_interface_objects:
                 sorting_interface = self.data_interface_objects[
                     f"Sorting{probe_name}"
                 ]
-                
+
                 # Sanity check no sorted spikes are outside recording range
                 exceeded_spikes = waveform_tools.has_exceeding_spikes(
                     recording=recording_interface.recording_extractor,
@@ -113,7 +113,7 @@ def temporally_align_data_interfaces(self):
                     raise ValueError(
                         f"Spikes exceeding recording found in Sorting{probe_name}!"
                     )
-                    
+
                 # Register recording
                 sorting_interface.register_recording(recording_interface)