make _get_preferred_store private

skypilot-org · Mar 25, 2024 · c614241 · c614241
1 parent 60fe268
commit c614241
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sky/task.py b/sky/task.py
@@ -855,7 +855,7 @@ def update_storage_mounts(
         task_storage_mounts.update(storage_mounts)
         return self.set_storage_mounts(task_storage_mounts)
 
-    def get_preferred_store(
+    def _get_preferred_store(
             self) -> Tuple[storage_lib.StoreType, Optional[str]]:
         """Returns the preferred store type and region for this task."""
         # TODO(zhwu, romilb): The optimizer should look at the source and
@@ -906,7 +906,7 @@ def sync_storage_mounts(self) -> None:
         """
         for storage in self.storage_mounts.values():
             if len(storage.stores) == 0:
-                store_type, store_region = self.get_preferred_store()
+                store_type, store_region = self._get_preferred_store()
                 self.storage_plans[storage] = store_type
                 storage.add_store(store_type, store_region)
             else: