Fix formatting and imports

src/spyglass/decoding/v1/core.py

@@ -1,16 +1,17 @@
 import numpy as np
 import pandas as pd
-from spyglass.common.common_behav import RawPosition, PositionIntervalMap
-from spyglass.common.common_interval import (
-    IntervalList,
-    interval_list_intersect,
-)
-from replay_trajectory_classification.observation_model import ObservationModel
 from replay_trajectory_classification.continuous_state_transitions import (
     RandomWalk,
     Uniform,
 )
 from replay_trajectory_classification.environments import Environment
+from replay_trajectory_classification.observation_model import ObservationModel
+
+from spyglass.common.common_behav import PositionIntervalMap, RawPosition
+from spyglass.common.common_interval import (
+    IntervalList,
+    interval_list_intersect,
+)
 
 
 def get_valid_ephys_position_times_from_interval(

src/spyglass/decoding/v1/dj_decoder_conversion.py

@@ -2,6 +2,8 @@
 so that datajoint can store them in tables."""
 
 
+import datajoint as dj
+from non_local_detector import ContFragClusterlessClassifier
 from non_local_detector.continuous_state_transitions import (
     Discrete,
     EmpiricalMovement,
@@ -17,15 +19,10 @@
     DiscreteStationaryDiagonal,
 )
 from non_local_detector.environment import Environment
-from non_local_detector.initial_conditions import (
-    UniformInitialConditions,
-)
+from non_local_detector.initial_conditions import UniformInitialConditions
 from non_local_detector.observation_models import ObservationModel
 from track_linearization import make_track_graph
 
-import datajoint as dj
-from non_local_detector import ContFragClusterlessClassifier
-
 from spyglass.decoding.v1.dj_decoder_conversion import (
     convert_classes_to_dict,
     restore_classes,
@@ -34,8 +31,9 @@
 schema = dj.schema("decoding_clusterless_v1")
 
 
-from non_local_detector.environment import Environment
-from track_linearization import make_track_graph
+import copy
+
+from non_local_detector import ContFragClusterlessClassifier
 from non_local_detector.continuous_state_transitions import (
     Discrete,
     EmpiricalMovement,
@@ -51,12 +49,9 @@
     DiscreteStationaryDiagonal,
 )
 from non_local_detector.environment import Environment
-from non_local_detector.initial_conditions import (
-    UniformInitialConditions,
-)
+from non_local_detector.initial_conditions import UniformInitialConditions
 from non_local_detector.observation_models import ObservationModel
-from non_local_detector import ContFragClusterlessClassifier
-import copy
+from track_linearization import make_track_graph
 
 
 def _convert_dict_to_class(d: dict, class_conversion: dict) -> object:

src/spyglass/decoding/v1/sorted_spikes.py

@@ -24,22 +24,23 @@
 from replay_trajectory_classification.initial_conditions import (
     UniformInitialConditions,
 )
-from spyglass.common.common_interval import IntervalList
-from spyglass.common.common_nwbfile import AnalysisNwbfile
-from spyglass.common.common_position import IntervalPositionInfo
-from spyglass.utils.dj_helper_fn import fetch_nwb
+
 from spyglass.common.common_behav import (
     convert_epoch_interval_name_to_position_interval_name,
 )
-from spyglass.decoding.core import (
+from spyglass.common.common_interval import IntervalList
+from spyglass.common.common_nwbfile import AnalysisNwbfile
+from spyglass.common.common_position import IntervalPositionInfo
+from spyglass.decoding.v1.core import (
     convert_valid_times_to_slice,
     get_valid_ephys_position_times_by_epoch,
 )
-from spyglass.decoding.dj_decoder_conversion import (
+from spyglass.decoding.v1.dj_decoder_conversion import (
     convert_classes_to_dict,
     restore_classes,
 )
 from spyglass.spikesorting.spikesorting_curation import CuratedSpikeSorting
+from spyglass.utils.dj_helper_fn import fetch_nwb
 
 schema = dj.schema("decoding_sortedspikes")
 

src/spyglass/decoding/v1/waveform_features.py

@@ -1,20 +1,21 @@
 import os
 import shutil
 import uuid
+from pathlib import Path
+
+import datajoint as dj
 import numpy as np
 import pandas as pd
-import datajoint as dj
 import pynwb
 import spikeinterface as si
-from pathlib import Path
+
+from spyglass.common.common_nwbfile import AnalysisNwbfile
 from spyglass.spikesorting import (
-    SpikeSortingOutput,
-    CurationV1,
     CuratedSpikeSortingV1,
+    CurationV1,
+    SpikeSortingOutput,
 )
-from spyglass.common.common_nwbfile import AnalysisNwbfile
 from spyglass.utils.dj_helper_fn import fetch_nwb
-from spyglass.spikesorting.v1.metric_utils import get_peak_channel
 
 schema = dj.schema("waveform_features")