Drop evaluation_id after adding merit values col

equinor · Nov 13, 2024 · 0606e09 · 0606e09
1 parent 0b99e8b
commit 0606e09
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/everest/everest_storage.py b/src/everest/everest_storage.py
@@ -459,7 +459,9 @@ def _handle_finished_batch_event(self, event: Event):
                 if merit_df is None:
                     objective_df = objective_df.drop("evaluation_id")
                 else:
-                    objective_df = objective_df.join(merit_df, on="evaluation_id")
+                    objective_df = objective_df.join(merit_df, on="evaluation_id").drop(
+                        "evaluation_id"
+                    )
 
             self._dataframes.batches.append(
                 BatchDataFrames(