Merge pull request #51 from swisstopo/fix/duplicate_column_in_documen…

…t_level_metrics Fix duplicate column 'document_name' in document_level_metrics table.
swisstopo · May 28, 2024 · 5b4c405 · 5b4c405 · github-actions · May 28, 2024
2 parents d26fbf7 + 015772f
commit 5b4c405
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