From 2d16562d8a14d7005919b66f163e5fa0394afd96 Mon Sep 17 00:00:00 2001 From: "Yngve S. Kristiansen" Date: Tue, 7 Jan 2025 12:54:18 +0100 Subject: [PATCH] Fixup data api logic --- src/everest/api/everest_data_api.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/everest/api/everest_data_api.py b/src/everest/api/everest_data_api.py index 352be213ff2..d15dd54a95c 100644 --- 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 (