Update init

khl02007 · Oct 10, 2023 · f70f2b0 · f70f2b0
1 parent 4ae1bb9
commit f70f2b0
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/src/spyglass/spikesorting/v1/__init__.py b/src/spyglass/spikesorting/v1/__init__.py
@@ -4,5 +4,18 @@
     SpikeSortingRecordingSelection,
     SpikeSortingRecording,
 )
-
-from .sorting import SpikeSorterParameter, SpikeSortingSelection, SpikeSortingV1
+from .artifact import (
+    ArtifactDetectionParameter,
+    ArtifactDetectionSelection,
+    ArtifactRemovedInterval,
+)
+from .sorting import SpikeSorterParameter, SpikeSortingSelection, SpikeSorting
+from .curation import CurationV1
+from .metric_curation import (
+    WaveformParameter,
+    MetricParameter,
+    MetricCurationParameter,
+    MetricCurationSelection,
+    MetricCuration,
+)
+from .figurl_curation import FigURLCurationSelection, FigURLCuration