Fix duplicate column 'document_name' in document_level_metrics table.

swisstopo · May 28, 2024 · 015772f · 015772f
1 parent d26fbf7
commit 015772f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/stratigraphy/benchmark/score.py b/src/stratigraphy/benchmark/score.py
@@ -146,7 +146,9 @@ def evaluate_borehole_extraction(predictions: dict, number_of_truth_values: dict
     layer_metrics, layer_document_level_metrics = evaluate_layer_extraction(predictions, number_of_truth_values)
     coordinate_metrics, coordinate_document_level_metrics = evaluate_metadata(predictions)
     metrics = {**layer_metrics, **coordinate_metrics}
-    document_level_metrics = pd.concat([layer_document_level_metrics, coordinate_document_level_metrics], axis=1)
+    document_level_metrics = pd.merge(
+        layer_document_level_metrics, coordinate_document_level_metrics, on="document_name"
+    )
     return metrics, document_level_metrics