diff --git a/src/runloop_api_client/resources/scenarios/scenarios.py b/src/runloop_api_client/resources/scenarios/scenarios.py index 81a4a9db4..65232bc46 100644 --- a/src/runloop_api_client/resources/scenarios/scenarios.py +++ b/src/runloop_api_client/resources/scenarios/scenarios.py @@ -460,6 +460,7 @@ def start_run_and_await_env_ready( benchmark_run_id: Optional[str] | NotGiven = NOT_GIVEN, metadata: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN, run_name: Optional[str] | NotGiven = NOT_GIVEN, + run_profile: Optional[scenario_start_run_params.RunProfile] | NotGiven = NOT_GIVEN, polling_config: PollingConfig | None = None, # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs. # The extra values given here take precedence over values defined on the client or passed to this method. @@ -475,6 +476,7 @@ def start_run_and_await_env_ready( scenario_id: ID of the Scenario to run benchmark_run_id: Benchmark to associate the run run_name: Display name of the run + run_profile: Runtime configuration to use for this benchmark run polling_config: Optional polling configuration extra_headers: Send extra headers extra_query: Add additional query parameters to the request @@ -494,6 +496,7 @@ def start_run_and_await_env_ready( benchmark_run_id=benchmark_run_id, metadata=metadata, run_name=run_name, + run_profile=run_profile, extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, @@ -913,6 +916,7 @@ async def start_run_and_await_env_ready( benchmark_run_id: Optional[str] | NotGiven = NOT_GIVEN, metadata: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN, run_name: Optional[str] | NotGiven = NOT_GIVEN, + run_profile: Optional[scenario_start_run_params.RunProfile] | NotGiven = NOT_GIVEN, polling_config: PollingConfig | None = None, ) -> ScenarioRunView: """Start a new ScenarioRun and wait for its environment to be ready. @@ -921,6 +925,7 @@ async def start_run_and_await_env_ready( scenario_id: ID of the Scenario to run benchmark_run_id: Benchmark to associate the run run_name: Display name of the run + run_profile: Runtime configuration to use for this benchmark run polling_config: Optional polling configuration Returns: @@ -935,6 +940,7 @@ async def start_run_and_await_env_ready( benchmark_run_id=benchmark_run_id, metadata=metadata, run_name=run_name, + run_profile=run_profile, ) await self._client.devboxes.await_running(