Merge pull request #2987 from yger/patch_unit_location

Patch unit location
SpikeInterface · Jun 7, 2024 · 1393b2b · 1393b2b
2 parents c49acd0 + 4a0cc1d
commit 1393b2b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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