|
43 | 43 | import warnings |
44 | 44 | import xml.etree.ElementTree as ET |
45 | 45 |
|
46 | | -from OMPython.OMCSession import OMCSessionException, OMCSessionRunData, OMCSessionZMQ, OMCProcessLocal, OMCPath |
| 46 | +from OMPython.OMCSession import OMCSessionException, OMCSessionRunData, OMCSessionZMQ, OMCProcess, OMCPath |
47 | 47 |
|
48 | 48 | # define logger using the current module name as ID |
49 | 49 | logger = logging.getLogger(__name__) |
@@ -262,7 +262,9 @@ def definition(self) -> OMCSessionRunData: |
262 | 262 | cmd_timeout=self._timeout, |
263 | 263 | ) |
264 | 264 |
|
265 | | - omc_run_data_updated = self._session.omc_run_data_update(omc_run_data=omc_run_data) |
| 265 | + omc_run_data_updated = self._session.omc_run_data_update( |
| 266 | + omc_run_data=omc_run_data, |
| 267 | + ) |
266 | 268 |
|
267 | 269 | return omc_run_data_updated |
268 | 270 |
|
@@ -315,7 +317,7 @@ def __init__( |
315 | 317 | variableFilter: Optional[str] = None, |
316 | 318 | customBuildDirectory: Optional[str | os.PathLike] = None, |
317 | 319 | omhome: Optional[str] = None, |
318 | | - omc_process: Optional[OMCProcessLocal] = None, |
| 320 | + omc_process: Optional[OMCProcess] = None, |
319 | 321 | build: bool = True, |
320 | 322 | ) -> None: |
321 | 323 | """Initialize, load and build a model. |
@@ -380,8 +382,6 @@ def __init__( |
380 | 382 | self._linearized_states: list[str] = [] # linearization states list |
381 | 383 |
|
382 | 384 | if omc_process is not None: |
383 | | - if not isinstance(omc_process, OMCProcessLocal): |
384 | | - raise ModelicaSystemError("Invalid (local) omc process definition provided!") |
385 | 385 | self._getconn = OMCSessionZMQ(omc_process=omc_process) |
386 | 386 | else: |
387 | 387 | self._getconn = OMCSessionZMQ(omhome=omhome) |
|
0 commit comments