From c6142416def0a40a4fce279e855df136cc323d48 Mon Sep 17 00:00:00 2001 From: Romil Bhardwaj Date: Mon, 25 Mar 2024 00:45:48 -0700 Subject: [PATCH] make _get_preferred_store private --- sky/task.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sky/task.py b/sky/task.py index bc26a49c9cf..59ff39830b1 100644 --- 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: