Protect dtype casting only when column is in column_name_to_column_dtype

SpikeInterface · Jan 10, 2025 · 33feca3 · 33feca3
1 parent 8353160
commit 33feca3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/spikeinterface/qualitymetrics/quality_metric_calculator.py b/src/spikeinterface/qualitymetrics/quality_metric_calculator.py
@@ -248,7 +248,8 @@ def _compute_metrics(self, sorting_analyzer, unit_ids=None, verbose=False, metri
         # we do this because the convert_dtypes infers the wrong types sometimes.
         # the actual types for columns can be found in column_name_to_column_dtype dictionary.
         for column in metrics.columns:
-            metrics[column] = metrics[column].astype(column_name_to_column_dtype[column])
+            if column in column_name_to_column_dtype:
+                metrics[column] = metrics[column].astype(column_name_to_column_dtype[column])
 
         return metrics