diff --git a/src/spyglass/spikesorting/v1/__init__.py b/src/spyglass/spikesorting/v1/__init__.py index 8f89d2837..85702d9dd 100644 --- 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