Skip to content

Commit 93c4db9

Browse files
committed
chore: add overloads to simultion run method
1 parent fa7f93a commit 93c4db9

File tree

1 file changed

+27
-0
lines changed

1 file changed

+27
-0
lines changed

cognite/client/_api/simulators/routines.py

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -226,6 +226,33 @@ def list(
226226
filter=routines_filter.dump(),
227227
)
228228

229+
@overload
230+
def run(
231+
self,
232+
*,
233+
routine_external_id: str,
234+
inputs: Sequence[SimulationInputOverride] | None = None,
235+
run_time: int | None = None,
236+
queue: bool | None = None,
237+
log_severity: Literal["Debug", "Information", "Warning", "Error"] | None = None,
238+
wait: bool = True,
239+
timeout: float = 60,
240+
) -> SimulationRun: ...
241+
242+
@overload
243+
def run(
244+
self,
245+
*,
246+
routine_revision_external_id: str,
247+
model_revision_external_id: str,
248+
inputs: Sequence[SimulationInputOverride] | None = None,
249+
run_time: int | None = None,
250+
queue: bool | None = None,
251+
log_severity: Literal["Debug", "Information", "Warning", "Error"] | None = None,
252+
wait: bool = True,
253+
timeout: float = 60,
254+
) -> SimulationRun: ...
255+
229256
def run(
230257
self,
231258
routine_external_id: str | None = None,

0 commit comments

Comments
 (0)