File tree Expand file tree Collapse file tree 3 files changed +6
-6
lines changed Expand file tree Collapse file tree 3 files changed +6
-6
lines changed Original file line number Diff line number Diff line change @@ -34,7 +34,7 @@ class SavingOptions(BaseModel):
34
34
class Result (BaseModel ):
35
35
model_config = ConfigDict (arbitrary_types_allowed = True , extra = "forbid" )
36
36
37
- datasets : dict [str , OptimizationResult ]
37
+ optimization_results : dict [str , OptimizationResult ]
38
38
experiments : dict [str , ExperimentModel ]
39
39
optimization_info : OptimizationInfo
40
40
initial_parameters : Parameters
@@ -65,7 +65,7 @@ def save(
65
65
66
66
data_path = path / "data"
67
67
data_path .mkdir (exist_ok = True )
68
- for label , data in self .datasets .items ():
68
+ for label , data in self .optimization_results .items ():
69
69
dataset_path = data_path / f"{ label } .{ options .data_format } "
70
70
result_dict ["data" ][label ] = str (dataset_path )
71
71
if options .data_filter is not None :
Original file line number Diff line number Diff line change @@ -83,7 +83,7 @@ def optimize(
83
83
optimization_info = optimization_info , parameters = optimized_parameters
84
84
)
85
85
return Result (
86
- datasets = optimized_data ,
86
+ optimization_results = optimized_data ,
87
87
experiments = self .experiments ,
88
88
optimization_info = optimization_info ,
89
89
initial_parameters = parameters ,
Original file line number Diff line number Diff line change @@ -432,13 +432,13 @@ def create_clp_guide_dataset(
432
432
"""
433
433
if isinstance (result , xr .Dataset ):
434
434
dataset = result
435
- elif dataset_name is None or dataset_name not in result .datasets :
435
+ elif dataset_name is None or dataset_name not in result .optimization_results :
436
436
raise ValueError (
437
437
f"Unknown dataset { dataset_name !r} . "
438
- f"Known datasets are:\n { list (result .datasets .keys ())} "
438
+ f"Known datasets are:\n { list (result .optimization_results .keys ())} "
439
439
)
440
440
else :
441
- dataset = result .datasets [dataset_name ]
441
+ dataset = result .optimization_results [dataset_name ]
442
442
if clp_label not in dataset .clp_label :
443
443
raise ValueError (
444
444
f"Unknown clp_label { clp_label !r} . "
You can’t perform that action at this time.
0 commit comments