WIP: add missing import

LorenFrankLab · Jan 9, 2025 · a6f26f5 · a6f26f5
1 parent c028c70
commit a6f26f5
Showing 1 changed file with 1 addition and 10 deletions.
diff --git a/src/spyglass/spikesorting/v1/burst_curation.py b/src/spyglass/spikesorting/v1/burst_curation.py
@@ -12,16 +12,7 @@
 )
 
 from spyglass.decoding.utils import _get_peak_amplitude
-
-# from spyglass.spikesorting.v0.spikesorting_curation import (
-#     CuratedSpikeSorting,
-#     CuratedSpikeSortingSelection,
-#     Curation,
-#     Waveforms,
-#     WaveformSelection,
-# )
-# from spyglass.spikesorting.v0.spikesorting_sorting import SpikeSorting
-from spyglass.spikesorting.v1.metric_curation import MetricCuration
+from spyglass.spikesorting.v1.metric_curation import CurationV1, MetricCuration
 from spyglass.utils import logger
 
 schema = dj.schema("burst_v1")  # TODO: rename to spikesorting_burst_v1