diff --git a/release/ray_release/job_manager/anyscale_job_manager.py b/release/ray_release/job_manager/anyscale_job_manager.py index 65a8e7798e55..5868cc5922f3 100644 --- a/release/ray_release/job_manager/anyscale_job_manager.py +++ b/release/ray_release/job_manager/anyscale_job_manager.py @@ -7,6 +7,7 @@ from anyscale.sdk.anyscale_client.models import ( CreateProductionJob, + CreateProductionJobConfig, HaJobStates, ) from anyscale.controllers.job_controller import JobController, terminal_state @@ -71,7 +72,7 @@ def _run_job( name=self.cluster_manager.cluster_name, description=f"Smoke test: {self.cluster_manager.smoke_test}", project_id=self.cluster_manager.project_id, - config=dict( + config=CreateProductionJobConfig( entrypoint=full_cmd, runtime_env=runtime_env, build_id=self.cluster_manager.cluster_env_build_id,