Skip to content

Commit 33feca3

Browse files
committed
Protect dtype casting only when column is in column_name_to_column_dtype
1 parent 8353160 commit 33feca3

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

src/spikeinterface/qualitymetrics/quality_metric_calculator.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -248,7 +248,8 @@ def _compute_metrics(self, sorting_analyzer, unit_ids=None, verbose=False, metri
248248
# we do this because the convert_dtypes infers the wrong types sometimes.
249249
# the actual types for columns can be found in column_name_to_column_dtype dictionary.
250250
for column in metrics.columns:
251-
metrics[column] = metrics[column].astype(column_name_to_column_dtype[column])
251+
if column in column_name_to_column_dtype:
252+
metrics[column] = metrics[column].astype(column_name_to_column_dtype[column])
252253

253254
return metrics
254255

0 commit comments

Comments
 (0)