Skip to content

Commit ae68c42

Browse files
committed
[ModelicaSystemDoE] rename class constants
1 parent d2a1afc commit ae68c42

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
@@ -1581,8 +1581,8 @@ def run_doe():
15811581
15821582
"""
15831583

1584-
DF_COLUMNS_RESULTFILENAME: str = 'resultfilename'
1585-
DF_COLUMNS_RESULTS_AVAILABLE: str = 'results available'
1584+
DF_COLUMNS_RESULT_FILENAME: str = 'result filename'
1585+
DF_COLUMNS_RESULT_AVAILABLE: str = 'result available'
15861586

15871587
def __init__(
15881588
self,
@@ -1710,7 +1710,7 @@ def prepare(self) -> int:
17101710
{
17111711
'ID structure': idx_pc_structure,
17121712
'ID simple': idx_pc_simple,
1713-
self.DF_COLUMNS_RESULTFILENAME: resfilename,
1713+
self.DF_COLUMNS_RESULT_FILENAME: resfilename,
17141714
'structural parameters ID': idx_pc_structure,
17151715
}
17161716
| sim_param_structure
@@ -1719,7 +1719,7 @@ def prepare(self) -> int:
17191719
}
17201720
| sim_param_simple
17211721
| {
1722-
self.DF_COLUMNS_RESULTS_AVAILABLE: False,
1722+
self.DF_COLUMNS_RESULT_AVAILABLE: False,
17231723
}
17241724
)
17251725

@@ -1808,14 +1808,14 @@ def worker(worker_id, task_queue):
18081808
thread.join()
18091809

18101810
for row in self._sim_df.to_dict('records'):
1811-
resultfilename = row[self.DF_COLUMNS_RESULTFILENAME]
1811+
resultfilename = row[self.DF_COLUMNS_RESULT_FILENAME]
18121812
resultfile = self._resultpath / resultfilename
18131813

18141814
if resultfile.exists():
1815-
mask = self._sim_df[self.DF_COLUMNS_RESULTFILENAME] == resultfilename
1816-
self._sim_df.loc[mask, self.DF_COLUMNS_RESULTS_AVAILABLE] = True
1815+
mask = self._sim_df[self.DF_COLUMNS_RESULT_FILENAME] == resultfilename
1816+
self._sim_df.loc[mask, self.DF_COLUMNS_RESULT_AVAILABLE] = True
18171817

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

18211821
return sim_df_total == sim_df_done
@@ -1836,17 +1836,17 @@ def get_solutions(
18361836
if self._sim_df is None:
18371837
return None
18381838

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

18421842
if var_list is None:
1843-
resultfilename = self._sim_df[self.DF_COLUMNS_RESULTFILENAME].values[0]
1843+
resultfilename = self._sim_df[self.DF_COLUMNS_RESULT_FILENAME].values[0]
18441844
resultfile = self._resultpath / resultfilename
18451845
return self._mod.getSolutions(resultfile=resultfile)
18461846

18471847
sol_dict: dict[str, pd.DataFrame | str] = {}
18481848
for row in self._sim_df.to_dict('records'):
1849-
resultfilename = row[self.DF_COLUMNS_RESULTFILENAME]
1849+
resultfilename = row[self.DF_COLUMNS_RESULT_FILENAME]
18501850
resultfile = self._resultpath / resultfilename
18511851

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