[Release] Fix Anyscale API call (ray-project#33644)

Fixes API call broken due to anyscale==0.5.92. Signed-off-by: Antoni Baum <[email protected]>
elliottower · Mar 23, 2023 · 24a89a6 · 24a89a6
1 parent 231a1d8
commit 24a89a6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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,