Fix additional locations selection

dstackai · Aug 14, 2023 · 9f6339b · 9f6339b
1 parent 1b9a0d9
commit 9f6339b
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 5 deletions.
diff --git a/cli/dstack/_internal/hub/services/backends/azure/configurator.py b/cli/dstack/_internal/hub/services/backends/azure/configurator.py
@@ -145,7 +145,8 @@ def configure_backend(
             selected=backend_config.storage_account
         )
         backend_values.extra_locations = self._get_extra_locations_element(
-            location=backend_config.location
+            location=backend_config.location,
+            selected_locations=backend_config.extra_locations or [],
         )
         return backend_values
 
@@ -313,13 +314,15 @@ def _get_storage_account_element(self, selected: Optional[str]) -> BackendElemen
             element.selected = storage_accounts[0]
         return element
 
-    def _get_extra_locations_element(self, location: str) -> BackendMultiElement:
+    def _get_extra_locations_element(
+        self, location: str, selected_locations: List[str]
+    ) -> BackendMultiElement:
         element = BackendMultiElement()
         for l in LOCATION_VALUES:
             if l == location:
                 continue
             element.values.append(BackendElementValue(value=l, label=l))
-            element.selected.append(l)
+        element.selected = selected_locations
         return element
 
     def _get_resource_group(self) -> str:

diff --git a/cli/dstack/_internal/hub/services/backends/gcp/configurator.py b/cli/dstack/_internal/hub/services/backends/gcp/configurator.py
@@ -161,6 +161,7 @@ def configure_backend(
         backend_values.extra_regions = self._get_hub_extra_regions_element(
             region=backend_values.region.selected,
             region_names=list(regions.keys()),
+            selected_regions=backend_config.extra_regions or [],
         )
         return backend_values
 
@@ -350,14 +351,17 @@ def _get_hub_vpc_subnet_element(
         return element
 
     def _get_hub_extra_regions_element(
-        self, region: str, region_names: List[str]
+        self,
+        region: str,
+        region_names: List[str],
+        selected_regions: List[str],
     ) -> BackendMultiElement:
         element = BackendMultiElement()
         for region_name in region_names:
             if region_name == region:
                 continue
             element.values.append(BackendElementValue(value=region_name, label=region_name))
-            element.selected.append(region_name)
+        element.selected = selected_regions
         return element
 
     def _auth(self, credentials_data: Dict):