diff --git a/src/spyglass/spikesorting/v0/spikesorting_curation.py b/src/spyglass/spikesorting/v0/spikesorting_curation.py index 77a4e8edb..ba4d9d9de 100644 --- a/src/spyglass/spikesorting/v0/spikesorting_curation.py +++ b/src/spyglass/spikesorting/v0/spikesorting_curation.py @@ -604,11 +604,12 @@ def _compute_metric(self, waveform_extractor, metric_name, **metric_params): is_nn_iso = metric_name == "nn_isolation" is_nn_overlap = metric_name == "nn_noise_overlap" + is_num_spikes = metric_name == "num_spikes" min_spikes = metric_params.get("min_spikes", 10) for unit_id in waveform_extractor.sorting.get_unit_ids(): # checks to avoid bug in spikeinterface 0.98.2 - if num_spikes[unit_id] < min_spikes: + if num_spikes[unit_id] < min_spikes and not is_num_spikes: if is_nn_iso: metric[str(unit_id)] = (np.nan, np.nan) elif is_nn_overlap: