diff --git a/src/non_local_detector/models/nospike_cont_frag_model.py b/src/non_local_detector/models/nospike_cont_frag_model.py index 30ad10a..c6c861c 100644 --- a/src/non_local_detector/models/nospike_cont_frag_model.py +++ b/src/non_local_detector/models/nospike_cont_frag_model.py @@ -98,7 +98,7 @@ ] -class ContFragNoSpikeSortedSpikesClassifier(SortedSpikesDetector): +class NoSpikeContFragSortedSpikesClassifier(SortedSpikesDetector): def __init__( self, discrete_initial_conditions: np.ndarray = discrete_initial_conditions, @@ -143,7 +143,7 @@ def get_conditional_non_local_posterior(results): return acausal_posterior / acausal_posterior.sum("position") -class ContFragNoSpikeClusterlessClassifier(ClusterlessDetector): +class NoSpikeContFragClusterlessClassifier(ClusterlessDetector): def __init__( self, discrete_initial_conditions: np.ndarray = discrete_initial_conditions,