anyscale 0.26.33__py3-none-any.whl → 0.26.35__py3-none-any.whl

anyscale/_private/anyscale_client/anyscale_client.py

@@ -640,6 +640,7 @@ class AnyscaleClient(AnyscaleClientInterface):
     def get_compute_config_id(
         self,
         compute_config_name: Optional[str] = None,
+        cloud: Optional[str] = None,
         *,
         include_archived: bool = False,
     ) -> Optional[str]:
@@ -648,6 +649,11 @@ class AnyscaleClient(AnyscaleClientInterface):
             if version is None:
                 # Setting `version=-1` will return only the latest version if there are multiple.
                 version = -1
+
+            cloud_id = (
+                self.get_cloud_id(cloud_name=cloud) if cloud is not None else None
+            )
+
             cluster_computes = self._internal_api_client.search_compute_templates_api_v2_compute_templates_search_post(
                 ComputeTemplateQuery(
                     orgwide=True,
@@ -657,6 +663,7 @@ class AnyscaleClient(AnyscaleClientInterface):
                     if include_archived
                     else ArchiveStatus.NOT_ARCHIVED,
                     version=version,
+                    cloud_id=cloud_id,
                 )
             ).results
 

anyscale/_private/anyscale_client/common.py

@@ -200,7 +200,11 @@ class AnyscaleClientInterface(ABC):
 
     @abstractmethod
     def get_compute_config_id(
-        self, compute_config_name: Optional[str] = None, *, include_archived=False
+        self,
+        compute_config_name: Optional[str] = None,
+        cloud: Optional[str] = None,
+        *,
+        include_archived=False,
     ) -> Optional[str]:
         """Get the compute config ID for the provided name.
 

anyscale/_private/anyscale_client/fake_anyscale_client.py

@@ -468,7 +468,11 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         return None
 
     def get_compute_config_id(
-        self, compute_config_name: Optional[str] = None, *, include_archived=False
+        self,
+        compute_config_name: Optional[str] = None,
+        cloud: Optional[str] = None,
+        *,
+        include_archived=False,
     ) -> Optional[str]:
         if compute_config_name is not None:
             name, version = parse_cluster_compute_name_version(compute_config_name)
@@ -491,7 +495,8 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         if workspace_cluster is not None:
             return workspace_cluster.cluster_compute_id
 
-        return self.get_default_compute_config(cloud_id=self.get_cloud_id()).id
+        cloud_id = self.get_cloud_id(cloud_name=cloud)
+        return self.get_default_compute_config(cloud_id=cloud_id).id
 
     def archive_compute_config(self, *, compute_config_id: str):
         archived_config = self._compute_configs.pop(compute_config_id)