Skip to content

Commit 0752f3a

Browse files
committed
[ModelicaSystemDoE] rename class constants
1 parent f1d866c commit 0752f3a

2 files changed

Lines changed: 12 additions & 12 deletions

File tree

OMPython/ModelicaSystem.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1801,8 +1801,8 @@ def run_doe():
18011801
18021802
"""
18031803

1804-
DF_COLUMNS_RESULTFILENAME: str = 'resultfilename'
1805-
DF_COLUMNS_RESULTS_AVAILABLE: str = 'results available'
1804+
DF_COLUMNS_RESULT_FILENAME: str = 'result filename'
1805+
DF_COLUMNS_RESULT_AVAILABLE: str = 'result available'
18061806

18071807
def __init__(
18081808
self,
@@ -1930,7 +1930,7 @@ def prepare(self) -> int:
19301930
{
19311931
'ID structure': idx_pc_structure,
19321932
'ID simple': idx_pc_simple,
1933-
self.DF_COLUMNS_RESULTFILENAME: resfilename,
1933+
self.DF_COLUMNS_RESULT_FILENAME: resfilename,
19341934
'structural parameters ID': idx_pc_structure,
19351935
}
19361936
| sim_param_structure
@@ -1939,7 +1939,7 @@ def prepare(self) -> int:
19391939
}
19401940
| sim_param_simple
19411941
| {
1942-
self.DF_COLUMNS_RESULTS_AVAILABLE: False,
1942+
self.DF_COLUMNS_RESULT_AVAILABLE: False,
19431943
}
19441944
)
19451945

@@ -2028,14 +2028,14 @@ def worker(worker_id, task_queue):
20282028
thread.join()
20292029

20302030
for row in self._sim_df.to_dict('records'):
2031-
resultfilename = row[self.DF_COLUMNS_RESULTFILENAME]
2031+
resultfilename = row[self.DF_COLUMNS_RESULT_FILENAME]
20322032
resultfile = self._resultpath / resultfilename
20332033

20342034
if resultfile.exists():
2035-
mask = self._sim_df[self.DF_COLUMNS_RESULTFILENAME] == resultfilename
2036-
self._sim_df.loc[mask, self.DF_COLUMNS_RESULTS_AVAILABLE] = True
2035+
mask = self._sim_df[self.DF_COLUMNS_RESULT_FILENAME] == resultfilename
2036+
self._sim_df.loc[mask, self.DF_COLUMNS_RESULT_AVAILABLE] = True
20372037

2038-
sim_df_done = self._sim_df[self.DF_COLUMNS_RESULTS_AVAILABLE].sum()
2038+
sim_df_done = self._sim_df[self.DF_COLUMNS_RESULT_AVAILABLE].sum()
20392039
logger.info(f"All workers finished ({sim_df_done} of {sim_df_total} simulations with a result file).")
20402040

20412041
return sim_df_total == sim_df_done
@@ -2056,17 +2056,17 @@ def get_solutions(
20562056
if self._sim_df is None:
20572057
return None
20582058

2059-
if self._sim_df.shape[0] == 0 or self._sim_df[self.DF_COLUMNS_RESULTS_AVAILABLE].sum() == 0:
2059+
if self._sim_df.shape[0] == 0 or self._sim_df[self.DF_COLUMNS_RESULT_AVAILABLE].sum() == 0:
20602060
raise ModelicaSystemError("No result files available - all simulations did fail?")
20612061

20622062
if var_list is None:
2063-
resultfilename = self._sim_df[self.DF_COLUMNS_RESULTFILENAME].values[0]
2063+
resultfilename = self._sim_df[self.DF_COLUMNS_RESULT_FILENAME].values[0]
20642064
resultfile = self._resultpath / resultfilename
20652065
return self._mod.getSolutions(resultfile=resultfile)
20662066

20672067
sol_dict: dict[str, pd.DataFrame | str] = {}
20682068
for row in self._sim_df.to_dict('records'):
2069-
resultfilename = row[self.DF_COLUMNS_RESULTFILENAME]
2069+
resultfilename = row[self.DF_COLUMNS_RESULT_FILENAME]
20702070
resultfile = self._resultpath / resultfilename
20712071

20722072
try:

tests/test_ModelicaSystemDoE.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ def test_ModelicaSystemDoE(tmp_path, model_doe, param_doe):
5959
assert df_doe['results available'].sum() == 16
6060

6161
for row in df_doe.to_dict('records'):
62-
resultfilename = row[mod_doe.DF_COLUMNS_RESULTFILENAME]
62+
resultfilename = row[mod_doe.DF_COLUMNS_RESULT_FILENAME]
6363
resultfile = mod_doe._resultpath / resultfilename
6464

6565
var_dict = {

0 commit comments

Comments
 (0)