From 8cb700fcf9e0b8cc3bcccae9e00cfabd7143e23d Mon Sep 17 00:00:00 2001 From: Shawn Pan Date: Fri, 24 Mar 2023 16:24:44 -0700 Subject: [PATCH] Revert "[Release] Fix Anyscale API call (#33644)" This reverts commit 24a89a6ec7d5bfd2a6d875b3ed8da79eaf7f288c. --- release/ray_release/job_manager/anyscale_job_manager.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/release/ray_release/job_manager/anyscale_job_manager.py b/release/ray_release/job_manager/anyscale_job_manager.py index 5868cc5922f3..65a8e7798e55 100644 --- a/release/ray_release/job_manager/anyscale_job_manager.py +++ b/release/ray_release/job_manager/anyscale_job_manager.py @@ -7,7 +7,6 @@ from anyscale.sdk.anyscale_client.models import ( CreateProductionJob, - CreateProductionJobConfig, HaJobStates, ) from anyscale.controllers.job_controller import JobController, terminal_state @@ -72,7 +71,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=CreateProductionJobConfig( + config=dict( entrypoint=full_cmd, runtime_env=runtime_env, build_id=self.cluster_manager.cluster_env_build_id,