fix resource api to match skypilot 0.4.1

Trainy-ai · Nov 1, 2023 · 2ae50c7 · 2ae50c7
1 parent 6284c4c
commit 2ae50c7
Show file tree

Hide file tree

Showing 4 changed files with 268 additions and 1,210 deletions.
diff --git a/llm_atc/launch.py b/llm_atc/launch.py
@@ -100,6 +100,6 @@ def launch(self) -> sky.Task:
         resource = list(task.get_resources())[0]
         resource._set_accelerators(self.accelerator, None)
         resource._cloud = sky.clouds.CLOUD_REGISTRY.from_str(self.cloud)
-        resource._set_region_zone(self.region, self.zone)
+        resource._validate_and_set_region_zone(self.region, self.zone)
         task.set_resources(resource)
         return task
diff --git a/llm_atc/serve.py b/llm_atc/serve.py
@@ -78,6 +78,6 @@ def serve(self) -> sky.Task:
         resource = list(serve_task.get_resources())[0]
         resource._set_accelerators(self.accelerator, None)
         resource._cloud = sky.clouds.CLOUD_REGISTRY.from_str(self.cloud)
-        resource._set_region_zone(self.region, self.zone)
+        resource._validate_and_set_region_zone(self.region, self.zone)
         serve_task.set_resources(resource)
         return serve_task