[UX] Optimizing error message when disabled cloud storage attempted

sky/data/storage.py

@@ -42,6 +42,8 @@
 # R2 to be an option as preferred store type
 STORE_ENABLED_CLOUDS = [clouds.AWS(), clouds.GCP()]
 
+STORE_TYPE_TO_CLOUD_TYPE = {'s3': 'AWS', 'gcs': 'GCP', 'r2': 'Cloudflare'}
+
 # Maximum number of concurrent rsync upload processes
 _MAX_CONCURRENT_UPLOADS = 32
 

sky/exceptions.py

@@ -209,3 +209,8 @@ class ClusterOwnerIdentityMismatchError(Exception):
 class NoCloudAccessError(Exception):
     """Raised when all clouds are disabled."""
     pass
+
+
+class CloudDisabledError(Exception):
+    """Raised when attempted cloud is disabled"""
+    pass

sky/task.py

@@ -17,6 +17,7 @@
 from sky.skylet import constants
 from sky.utils import schemas
 from sky.utils import ux_utils
+from sky.adaptors import cloudflare
 
 if typing.TYPE_CHECKING:
     from sky import resources as resources_lib
@@ -295,10 +296,24 @@ def from_yaml(yaml_path: str) -> 'Task':
 
         task_storage_mounts: Dict[str, storage_lib.Storage] = {}
         all_storages = fm_storages
+        enabled_clouds = global_user_state.get_enabled_clouds()
+        enabled_clouds = [str(cloud) for cloud in enabled_clouds]
+        r2_is_enabled, _ = cloudflare.check_credentials()
+        if r2_is_enabled:
+            enabled_clouds.append('r2')
         for storage in all_storages:
             mount_path = storage[0]
             assert mount_path, \
                 'Storage mount path cannot be empty.'
+            store_type = storage[1]['store']
+            if store_type:
+                cloud_type = storage_lib.STORE_TYPE_TO_CLOUD_TYPE[store_type]
+                if cloud_type not in enabled_clouds:
+                    with ux_utils.print_exception_no_traceback():
+                        raise exceptions.CloudDisabledError(
+                            f'Storage \'store:{store_type}\' specified, but '
+                            f'\'{cloud_type}\' access is disabled. Enable '
+                            f'\'{cloud_type}\' to fix.')
             try:
                 storage_obj = storage_lib.Storage.from_yaml_config(storage[1])
             except exceptions.StorageSourceError as e: