@@ -161,13 +161,15 @@ def get_opt_status(output_folder: str) -> dict[str, Any]:
161
161
objective_names = storage .data .objective_functions ["objective_name" ].to_list ()
162
162
control_names = storage .data .controls ["control_name" ].to_list ()
163
163
164
- expected_objectives = pl .concat (
165
- [
166
- b .batch_objectives .select (objective_names )
167
- for b in storage .data .batches
168
- if b .batch_objectives is not None
169
- ]
170
- ).to_dict (as_series = False )
164
+ objectives = [
165
+ b .batch_objectives .select (objective_names )
166
+ for b in storage .data .batches
167
+ if b .batch_objectives is not None
168
+ ]
169
+
170
+ expected_objectives = (
171
+ {} if not objectives else pl .concat (objectives ).to_dict (as_series = False )
172
+ )
171
173
172
174
expected_total_objective = [
173
175
b .batch_objectives ["total_objective_value" ].item ()
@@ -191,16 +193,18 @@ def get_opt_status(output_folder: str) -> dict[str, Any]:
191
193
"objective_value" : expected_total_objective ,
192
194
"expected_objectives" : expected_objectives ,
193
195
}
196
+ controls = [
197
+ b .realization_controls .select (control_names )
198
+ for b in storage .data .batches
199
+ if b .realization_controls is not None
200
+ ]
201
+ control_history = (
202
+ {} if not controls else pl .concat (controls ).to_dict (as_series = False )
203
+ )
194
204
195
205
return {
196
206
"objective_history" : expected_total_objective ,
197
- "control_history" : pl .concat (
198
- [
199
- b .realization_controls .select (control_names )
200
- for b in storage .data .batches
201
- if b .realization_controls is not None
202
- ]
203
- ).to_dict (as_series = False ),
207
+ "control_history" : control_history ,
204
208
"objectives_history" : expected_objectives ,
205
209
"accepted_control_indices" : improvement_batches ,
206
210
"cli_monitor_data" : cli_monitor_data ,
0 commit comments