Make importing nospike model easier

LorenFrankLab · Aug 8, 2024 · c9157f1 · c9157f1
1 parent 905b5f4
commit c9157f1
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git 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
@@ -12,3 +12,7 @@
     NonLocalClusterlessDetector,
     NonLocalSortedSpikesDetector,
 )
+from non_local_detector.models.nospike_cont_frag_model import (  # noqa
+    NoSpikeContFragClusterlessClassifier,
+    NoSpikeContFragSortedSpikesClassifier,
+)
diff --git a/...ocal_detector/models/nospike_cont_frag.py → ...etector/models/nospike_cont_frag_model.py b/...ocal_detector/models/nospike_cont_frag.py → ...etector/models/nospike_cont_frag_model.py