Skip to content

Commit 5914f63

Browse files
committed
[ModelicaSystemDoE] rename class constants
1 parent 72532c3 commit 5914f63

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
@@ -1696,8 +1696,8 @@ def run_doe():
16961696
16971697
"""
16981698

1699-
DF_COLUMNS_RESULTFILENAME: str = 'resultfilename'
1700-
DF_COLUMNS_RESULTS_AVAILABLE: str = 'results available'
1699+
DF_COLUMNS_RESULT_FILENAME: str = 'result filename'
1700+
DF_COLUMNS_RESULT_AVAILABLE: str = 'result available'
17011701

17021702
def __init__(
17031703
self,
@@ -1825,7 +1825,7 @@ def prepare(self) -> int:
18251825
{
18261826
'ID structure': idx_pc_structure,
18271827
'ID simple': idx_pc_simple,
1828-
self.DF_COLUMNS_RESULTFILENAME: resfilename,
1828+
self.DF_COLUMNS_RESULT_FILENAME: resfilename,
18291829
'structural parameters ID': idx_pc_structure,
18301830
}
18311831
| sim_param_structure
@@ -1834,7 +1834,7 @@ def prepare(self) -> int:
18341834
}
18351835
| sim_param_simple
18361836
| {
1837-
self.DF_COLUMNS_RESULTS_AVAILABLE: False,
1837+
self.DF_COLUMNS_RESULT_AVAILABLE: False,
18381838
}
18391839
)
18401840

@@ -1923,14 +1923,14 @@ def worker(worker_id, task_queue):
19231923
thread.join()
19241924

19251925
for row in self._sim_df.to_dict('records'):
1926-
resultfilename = row[self.DF_COLUMNS_RESULTFILENAME]
1926+
resultfilename = row[self.DF_COLUMNS_RESULT_FILENAME]
19271927
resultfile = self._resultpath / resultfilename
19281928

19291929
if resultfile.exists():
1930-
mask = self._sim_df[self.DF_COLUMNS_RESULTFILENAME] == resultfilename
1931-
self._sim_df.loc[mask, self.DF_COLUMNS_RESULTS_AVAILABLE] = True
1930+
mask = self._sim_df[self.DF_COLUMNS_RESULT_FILENAME] == resultfilename
1931+
self._sim_df.loc[mask, self.DF_COLUMNS_RESULT_AVAILABLE] = True
19321932

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

19361936
return sim_df_total == sim_df_done
@@ -1951,17 +1951,17 @@ def get_solutions(
19511951
if self._sim_df is None:
19521952
return None
19531953

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

19571957
if var_list is None:
1958-
resultfilename = self._sim_df[self.DF_COLUMNS_RESULTFILENAME].values[0]
1958+
resultfilename = self._sim_df[self.DF_COLUMNS_RESULT_FILENAME].values[0]
19591959
resultfile = self._resultpath / resultfilename
19601960
return self._mod.getSolutions(resultfile=resultfile)
19611961

19621962
sol_dict: dict[str, pd.DataFrame | str] = {}
19631963
for row in self._sim_df.to_dict('records'):
1964-
resultfilename = row[self.DF_COLUMNS_RESULTFILENAME]
1964+
resultfilename = row[self.DF_COLUMNS_RESULT_FILENAME]
19651965
resultfile = self._resultpath / resultfilename
19661966

19671967
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)