From edcfde37c7c6e66cf1b345c44d22e8aa4357a4ba Mon Sep 17 00:00:00 2001
From: Romil Bhardwaj <romil.bhardwaj@gmail.com>
Date: Tue, 11 Oct 2022 11:50:29 -0700
Subject: [PATCH] [Storage] Add `StorageMode` to __init__ (#1223)

* Add storage mode to __init__

* fix
---
 sky/__init__.py      | 3 ++-
 sky/data/__init__.py | 4 ++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/sky/__init__.py b/sky/__init__.py
index 451bfe31e10..2f6d95e0d77 100644
--- 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
index 646e8fdd410..b0b97b8753b 100644
--- 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']