[Storage] Add StorageMode to __init__ (skypilot-org#1223)

* Add storage mode to __init__

* fix

Author: romilbhardwaj

Diff for: 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 for: 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']