Fixup data api logic

equinor · Jan 7, 2025 · 2d16562 · 2d16562
1 parent 47ddb2d
commit 2d16562
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/everest/api/everest_data_api.py b/src/everest/api/everest_data_api.py
@@ -144,6 +144,7 @@ def single_objective_values(self):
             ]
         )
         objectives = self._ever_storage.data.objective_functions
+        objective_names = objectives["objective_name"].unique().to_list()
 
         for o in objectives.to_dicts():
             batch_datas = batch_datas.with_columns(
@@ -154,7 +155,7 @@ def single_objective_values(self):
             "batch",
             "objective",
             "accepted",
-            *self._ever_storage.data.objective_functions["objective_name"].to_list(),
+            *(objective_names if len(objective_names) > 1 else []),
         ]
 
         return (