diff --git a/src/non_local_detector/__init__.py b/src/non_local_detector/__init__.py index 11bc616..f91577a 100644 --- a/src/non_local_detector/__init__.py +++ b/src/non_local_detector/__init__.py @@ -23,6 +23,8 @@ MultiEnvironmentSortedSpikesClassifier, NonLocalClusterlessDetector, NonLocalSortedSpikesDetector, + NoSpikeContFragClusterlessClassifier, + NoSpikeContFragSortedSpikesClassifier, SortedSpikesDecoder, ) diff --git a/src/non_local_detector/models/__init__.py b/src/non_local_detector/models/__init__.py index c72d4b5..6d786fa 100644 --- a/src/non_local_detector/models/__init__.py +++ b/src/non_local_detector/models/__init__.py @@ -12,3 +12,7 @@ NonLocalClusterlessDetector, NonLocalSortedSpikesDetector, ) +from non_local_detector.models.nospike_cont_frag_model import ( # noqa + NoSpikeContFragClusterlessClassifier, + NoSpikeContFragSortedSpikesClassifier, +) diff --git a/src/non_local_detector/models/nospike_cont_frag.py b/src/non_local_detector/models/nospike_cont_frag_model.py similarity index 100% rename from src/non_local_detector/models/nospike_cont_frag.py rename to src/non_local_detector/models/nospike_cont_frag_model.py