|
27 | 27 | ) |
28 | 28 | from ..._base_client import make_request_options |
29 | 29 | from ...types.beta.task_group import TaskGroup |
| 30 | +from ...types.run_input_param import RunInputParam |
30 | 31 | from ...types.task_spec_param import TaskSpecParam |
31 | 32 | from ...types.beta.parallel_beta_param import ParallelBetaParam |
32 | | -from ...types.beta.beta_run_input_param import BetaRunInputParam |
33 | 33 | from ...types.beta.task_group_run_response import TaskGroupRunResponse |
34 | 34 | from ...types.beta.task_group_events_response import TaskGroupEventsResponse |
35 | 35 | from ...types.beta.task_group_get_runs_response import TaskGroupGetRunsResponse |
@@ -140,7 +140,7 @@ def add_runs( |
140 | 140 | self, |
141 | 141 | task_group_id: str, |
142 | 142 | *, |
143 | | - inputs: Iterable[BetaRunInputParam], |
| 143 | + inputs: Iterable[RunInputParam], |
144 | 144 | default_task_spec: Optional[TaskSpecParam] | Omit = omit, |
145 | 145 | betas: List[ParallelBetaParam] | Omit = omit, |
146 | 146 | # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs. |
@@ -427,7 +427,7 @@ async def add_runs( |
427 | 427 | self, |
428 | 428 | task_group_id: str, |
429 | 429 | *, |
430 | | - inputs: Iterable[BetaRunInputParam], |
| 430 | + inputs: Iterable[RunInputParam], |
431 | 431 | default_task_spec: Optional[TaskSpecParam] | Omit = omit, |
432 | 432 | betas: List[ParallelBetaParam] | Omit = omit, |
433 | 433 | # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs. |
|
0 commit comments