Fixing bug

SpikeInterface · Jun 27, 2024 · 34a1bc2 · 34a1bc2
1 parent b9c2638
commit 34a1bc2
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/spikeinterface/core/analyzer_extension_core.py b/src/spikeinterface/core/analyzer_extension_core.py
@@ -597,9 +597,10 @@ def _merge_extension_data(
                         weights[count] = counts[id]
                     weights /= weights.sum()
                     new_data[key][unit_ind] = (arr[keep_unit_indices, :, :] * weights[:, np.newaxis, np.newaxis]).sum(0)
-                    chan_ids = new_sorting_analyzer.sparsity.unit_id_to_channel_indices[unit_id]
-                    mask = ~np.isin(np.arange(arr.shape[2]), chan_ids)
-                    new_data[key][unit_ind][:, mask] = 0
+                    if new_sorting_analyzer.sparsity is not None:
+                        chan_ids = new_sorting_analyzer.sparsity.unit_id_to_channel_indices[unit_id]
+                        mask = ~np.isin(np.arange(arr.shape[2]), chan_ids)
+                        new_data[key][unit_ind][:, mask] = 0
 
         return new_data