|
21 | 21 | import numpy as np |
22 | 22 |
|
23 | 23 | from OMPython.OMCSession import ( |
24 | | - ModelExecutionData, |
25 | 24 | ModelExecutionException, |
26 | | - |
27 | | - OMCPath, |
28 | | - OMCSession, |
29 | | - OMCSessionDummy, |
30 | | - OMCSessionException, |
| 25 | + ModelExecutionData, |
| 26 | + PathBase, |
| 27 | + SessionBase, |
| 28 | + SessionRunner, |
31 | 29 | OMCSessionLocal, |
| 30 | + OMCSessionException, |
32 | 31 | ) |
33 | 32 |
|
| 33 | +OMCPath = PathBase |
| 34 | +OMCSession = SessionBase |
| 35 | + |
34 | 36 | # define logger using the current module name as ID |
35 | 37 | logger = logging.getLogger(__name__) |
36 | 38 |
|
@@ -1665,8 +1667,6 @@ def buildModel(self, variableFilter: Optional[str] = None): |
1665 | 1667 | def sendExpression(self, expr: str, parsed: bool = True) -> Any: |
1666 | 1668 | """ |
1667 | 1669 | Wrapper for OMCSession.sendExpression(). |
1668 | | -
|
1669 | | - TODO: rename expr => command |
1670 | 1670 | """ |
1671 | 1671 | try: |
1672 | 1672 | retval = self._session.sendExpression(command=expr, parsed=parsed) |
@@ -2244,8 +2244,6 @@ def prepare(self) -> int: |
2244 | 2244 | build_dir.mkdir() |
2245 | 2245 | self._mod.setWorkDirectory(work_directory=build_dir) |
2246 | 2246 |
|
2247 | | - # TODO: check omc_mod |
2248 | | - |
2249 | 2247 | sim_param_structure = {} |
2250 | 2248 | for idx_structure, pk_structure in enumerate(param_structure.keys()): |
2251 | 2249 | sim_param_structure[pk_structure] = pc_structure[idx_structure] |
@@ -2423,7 +2421,6 @@ def worker(worker_id, task_queue): |
2423 | 2421 | def get_doe_solutions( |
2424 | 2422 | self, |
2425 | 2423 | var_list: Optional[list] = None, |
2426 | | - # TODO: omc_mod |
2427 | 2424 | ) -> Optional[tuple[str] | dict[str, dict[str, np.ndarray]]]: |
2428 | 2425 | """ |
2429 | 2426 | Get all solutions of the DoE run. The following return values are possible: |
@@ -2453,8 +2450,6 @@ def get_doe_solutions( |
2453 | 2450 | if len(self._doe_def) == 0: |
2454 | 2451 | raise ModelicaSystemError("No result files available - all simulations did fail?") |
2455 | 2452 |
|
2456 | | - # TODO: check omc_mod |
2457 | | - |
2458 | 2453 | sol_dict: dict[str, dict[str, Any]] = {} |
2459 | 2454 | for resultfilename in self._doe_def: |
2460 | 2455 | resultfile = self._resultpath / resultfilename |
@@ -2495,12 +2490,12 @@ class ModelicaSystemRunner(ModelicaSystemBase): |
2495 | 2490 | def __init__( |
2496 | 2491 | self, |
2497 | 2492 | work_directory: Optional[str | os.PathLike] = None, |
2498 | | - session: Optional[OMCSession] = None, |
| 2493 | + session: Optional[SessionBase] = None, |
2499 | 2494 | ) -> None: |
2500 | 2495 | if session is None: |
2501 | | - session = OMCSessionDummy() |
| 2496 | + session = SessionRunner() |
2502 | 2497 |
|
2503 | | - if not isinstance(session, OMCSessionDummy): |
| 2498 | + if not isinstance(session, SessionRunner): |
2504 | 2499 | raise ModelicaSystemError("Only working if OMCsessionDummy is used!") |
2505 | 2500 |
|
2506 | 2501 | super().__init__( |
|
0 commit comments