From 0f06a9e59809ae16c7d70bf458a9efa72daa61c1 Mon Sep 17 00:00:00 2001
From: Romil <romil.bhardwaj@gmail.com>
Date: Tue, 11 Oct 2022 10:32:51 -0700
Subject: [PATCH 1/2] Add storage mode to __init__

---
 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..adafce71119 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']

From 70b5bbf6847bd2cd94e163955eef6b027330b196 Mon Sep 17 00:00:00 2001
From: Romil <romil.bhardwaj@gmail.com>
Date: Tue, 11 Oct 2022 10:43:33 -0700
Subject: [PATCH 2/2] fix

---
 sky/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sky/__init__.py b/sky/__init__.py
index adafce71119..2f6d95e0d77 100644
--- a/sky/__init__.py
+++ b/sky/__init__.py
@@ -43,7 +43,7 @@
     'list_accelerators',
     '__root_dir__',
     'Storage',
-    'StorageMode'
+    'StorageMode',
     'StoreType',
     'ClusterStatus',
     'JobStatus',