diff --git a/src/spikeinterface/postprocessing/unit_localization.py b/src/spikeinterface/postprocessing/unit_localization.py index e40523e7e5..16d9955e58 100644 --- a/src/spikeinterface/postprocessing/unit_localization.py +++ b/src/spikeinterface/postprocessing/unit_localization.py @@ -239,7 +239,7 @@ def compute_monopolar_triangulation( contact_locations = sorting_analyzer.get_channel_locations() sparsity = compute_sparsity(sorting_analyzer, method="radius", radius_um=radius_um) - templates = get_dense_templates_array(sorting_analyzer) + templates = get_dense_templates_array(sorting_analyzer, return_scaled=sorting_analyzer.return_scaled) nbefore = _get_nbefore(sorting_analyzer) if enforce_decrease: @@ -303,7 +303,7 @@ def compute_center_of_mass(sorting_analyzer, peak_sign="neg", radius_um=75, feat assert feature in ["ptp", "mean", "energy", "peak_voltage"], f"{feature} is not a valid feature" sparsity = compute_sparsity(sorting_analyzer, peak_sign=peak_sign, method="radius", radius_um=radius_um) - templates = get_dense_templates_array(sorting_analyzer) + templates = get_dense_templates_array(sorting_analyzer, return_scaled=sorting_analyzer.return_scaled) nbefore = _get_nbefore(sorting_analyzer) unit_location = np.zeros((unit_ids.size, 2), dtype="float64") @@ -374,7 +374,7 @@ def compute_grid_convolution( contact_locations = sorting_analyzer.get_channel_locations() unit_ids = sorting_analyzer.unit_ids - templates = get_dense_templates_array(sorting_analyzer) + templates = get_dense_templates_array(sorting_analyzer, return_scaled=sorting_analyzer.return_scaled) nbefore = _get_nbefore(sorting_analyzer) nafter = templates.shape[1] - nbefore