[Storage] Add StorageMode to __init__ (#1223)

* Add storage mode to __init__ * fix
skypilot-org · Oct 11, 2022 · edcfde3 · edcfde3
1 parent 48e1e63
commit edcfde3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/sky/__init__.py b/sky/__init__.py
@@ -16,7 +16,7 @@
 from sky.resources import Resources
 from sky.task import Task
 from sky.optimizer import Optimizer, OptimizeTarget
-from sky.data import Storage, StoreType
+from sky.data import Storage, StorageMode, StoreType
 from sky.global_user_state import ClusterStatus
 from sky.skylet.job_lib import JobStatus
 from sky.core import (status, start, stop, down, autostop, queue, cancel,
@@ -43,6 +43,7 @@
     'list_accelerators',
     '__root_dir__',
     'Storage',
+    'StorageMode',
     'StoreType',
     'ClusterStatus',
     'JobStatus',

diff --git a/sky/data/__init__.py b/sky/data/__init__.py
@@ -1,4 +1,4 @@
 """Sky Data."""
-from sky.data.storage import Storage, StoreType
+from sky.data.storage import Storage, StorageMode, StoreType
 
-__all__ = ['Storage', 'StoreType']
+__all__ = ['Storage', 'StorageMode', 'StoreType']