skypilot-nightly 1.0.0.dev20250521__py3-none-any.whl → 1.0.0.dev20250522__py3-none-any.whl

sky/utils/admin_policy_utils.py

@@ -1,9 +1,8 @@
 """Admin policy utils."""
+import contextlib
 import copy
 import importlib
-import os
-import tempfile
-from typing import Optional, Tuple, Union
+from typing import Iterator, Optional, Tuple, Union
 
 import colorama
 
@@ -52,9 +51,31 @@ def _get_policy_cls(
     return policy_cls
 
 
+@contextlib.contextmanager
+def apply_and_use_config_in_current_request(
+    entrypoint: Union['dag_lib.Dag', 'task_lib.Task'],
+    request_options: Optional[admin_policy.RequestOptions] = None,
+) -> Iterator['dag_lib.Dag']:
+    """Applies an admin policy and override SkyPilot config for current request
+
+    This is a helper function of `apply()` that applies an admin policy and
+    overrides the SkyPilot config for the current request as a context manager.
+    The original SkyPilot config will be restored when the context manager is
+    exited.
+
+    Refer to `apply()` for more details.
+    """
+    original_config = skypilot_config.to_dict()
+    dag, mutated_config = apply(entrypoint, request_options)
+    if mutated_config != original_config:
+        with skypilot_config.replace_skypilot_config(mutated_config):
+            yield dag
+    else:
+        yield dag
+
+
 def apply(
     entrypoint: Union['dag_lib.Dag', 'task_lib.Task'],
-    use_mutated_config_in_current_request: bool = True,
     request_options: Optional[admin_policy.RequestOptions] = None,
 ) -> Tuple['dag_lib.Dag', config_utils.Config]:
     """Applies an admin policy (if registered) to a DAG or a task.
@@ -85,8 +106,7 @@ def apply(
         return dag, skypilot_config.to_dict()
 
     logger.info(f'Applying policy: {policy}')
-    original_config = skypilot_config.to_dict()
-    config = copy.deepcopy(original_config)
+    config = copy.deepcopy(skypilot_config.to_dict())
     mutated_dag = dag_lib.Dag()
     mutated_dag.name = dag.name
 
@@ -126,22 +146,6 @@ def apply(
         mutated_dag.graph.add_edge(mutated_dag.tasks[u_idx],
                                    mutated_dag.tasks[v_idx])
 
-    if (use_mutated_config_in_current_request and
-            original_config != mutated_config):
-        with tempfile.NamedTemporaryFile(
-                delete=False,
-                mode='w',
-                prefix='policy-mutated-skypilot-config-',
-                suffix='.yaml') as temp_file:
-
-            common_utils.dump_yaml(temp_file.name, dict(**mutated_config))
-            os.environ[skypilot_config.ENV_VAR_SKYPILOT_CONFIG] = temp_file.name
-            logger.debug(f'Updated SkyPilot config: {temp_file.name}')
-            # TODO(zhwu): This is not a clean way to update the SkyPilot config,
-            # because we are resetting the global context for a single DAG,
-            # which is conceptually weird.
-            importlib.reload(skypilot_config)
-
     logger.debug(f'Mutated user request: {mutated_user_request}')
     mutated_dag.policy_applied = True
     return mutated_dag, mutated_config

sky/utils/context.py

@@ -57,6 +57,7 @@ class Context(object):
         self._log_file = None
         self._log_file_handle = None
         self.env_overrides = {}
+        self.config_context = None
 
     def cancel(self):
         """Cancel the context."""
@@ -159,17 +160,25 @@ class ContextualEnviron(MutableMapping):
         ctx = get()
         if ctx is not None:
             if key in ctx.env_overrides:
-                return ctx.env_overrides[key]
+                value = ctx.env_overrides[key]
+                # None is used to indicate that the key is deleted in the
+                # context.
+                if value is None:
+                    raise KeyError(key)
+                return value
         return self._environ[key]
 
     def __iter__(self):
         ctx = get()
+        deleted_keys = set()
         if ctx is not None:
-            for key in ctx.env_overrides:
+            for key, value in ctx.env_overrides.items():
+                if value is None:
+                    deleted_keys.add(key)
                 yield key
             for key in self._environ:
                 # Deduplicate the keys
-                if key not in ctx.env_overrides:
+                if key not in ctx.env_overrides and key not in deleted_keys:
                     yield key
         else:
             return self._environ.__iter__()
@@ -178,10 +187,27 @@ class ContextualEnviron(MutableMapping):
         return len(dict(self))
 
     def __setitem__(self, key, value):
-        return self._environ.__setitem__(key, value)
+        ctx = get()
+        if ctx is not None:
+            ctx.env_overrides[key] = value
+        else:
+            self._environ.__setitem__(key, value)
 
     def __delitem__(self, key):
-        return self._environ.__delitem__(key)
+        ctx = get()
+        if ctx is not None:
+            if key in ctx.env_overrides:
+                del ctx.env_overrides[key]
+            elif key in self._environ:
+                # If the key is not set in the context but set in the environ
+                # of the process, we mark it as deleted in the context by
+                # setting the value to None.
+                ctx.env_overrides[key] = None
+            else:
+                # The key is not set in the context nor the process.
+                raise KeyError(key)
+        else:
+            self._environ.__delitem__(key)
 
     def __repr__(self):
         return self._environ.__repr__()
@@ -190,7 +216,11 @@ class ContextualEnviron(MutableMapping):
         copied = self._environ.copy()
         ctx = get()
         if ctx is not None:
-            copied.update(ctx.env_overrides)
+            for key in ctx.env_overrides:
+                if ctx.env_overrides[key] is None:
+                    copied.pop(key)
+                else:
+                    copied[key] = ctx.env_overrides[key]
         return copied
 
     def setdefault(self, key, default=None):

sky/utils/context_utils.py

@@ -1,5 +1,6 @@
 """Utilities for SkyPilot context."""
 import asyncio
+import contextvars
 import functools
 import io
 import multiprocessing
@@ -170,3 +171,17 @@ def cancellation_guard(func: F) -> F:
         return func(*args, **kwargs)
 
     return typing.cast(F, wrapper)
+
+
+# TODO(aylei): replace this with asyncio.to_thread once we drop support for
+# python 3.8
+def to_thread(func, /, *args, **kwargs):
+    """Asynchronously run function *func* in a separate thread.
+
+    This is same as asyncio.to_thread added in python 3.9
+    """
+    loop = asyncio.get_running_loop()
+    # This is critical to pass the current coroutine context to the new thread
+    pyctx = contextvars.copy_context()
+    func_call = functools.partial(pyctx.run, func, *args, **kwargs)
+    return loop.run_in_executor(None, func_call)

sky/utils/resources_utils.py

@@ -50,6 +50,20 @@ class DiskTier(enum.Enum):
         return types.index(self) <= types.index(other)
 
 
+class StorageType(enum.Enum):
+    """Storage type."""
+    # Durable network storage, e.g. GCP persistent disks
+    NETWORK = 'network'
+    # Local instance storage, e.g. GCP local SSDs
+    INSTANCE = 'instance'
+
+
+class DiskAttachMode(enum.Enum):
+    """Disk attach mode."""
+    READ_ONLY = 'read_only'
+    READ_WRITE = 'read_write'
+
+
 @dataclasses.dataclass
 class ClusterName:
     display_name: str

sky/utils/schemas.py

@@ -188,6 +188,35 @@ def _get_single_resources_schema():
                     }
                 }],
             },
+            'volumes': {
+                'type': 'array',
+                'items': {
+                    'type': 'object',
+                    'properties': {
+                        'disk_size': {
+                            'type': 'integer',
+                        },
+                        'disk_tier': {
+                            'type': 'string',
+                        },
+                        'path': {
+                            'type': 'string',
+                        },
+                        'auto_delete': {
+                            'type': 'boolean',
+                        },
+                        'storage_type': {
+                            'type': 'string',
+                        },
+                        'name': {
+                            'type': 'string',
+                        },
+                        'attach_mode': {
+                            'type': 'string',
+                        },
+                    },
+                },
+            },
             'disk_size': {
                 'type': 'integer',
             },
@@ -372,6 +401,23 @@ def get_storage_schema():
                     mode.value for mode in storage.StorageMode
                 ]
             },
+            'config': {
+                'type': 'object',
+                'properties': {
+                    'disk_size': {
+                        'type': 'integer',
+                    },
+                    'disk_tier': {
+                        'type': 'string',
+                    },
+                    'storage_type': {
+                        'type': 'string',
+                    },
+                    'attach_mode': {
+                        'type': 'string',
+                    },
+                },
+            },
             '_is_sky_managed': {
                 'type': 'boolean',
             },

{skypilot_nightly-1.0.0.dev20250521.dist-info → skypilot_nightly-1.0.0.dev20250522.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: skypilot-nightly
-Version: 1.0.0.dev20250521
+Version: 1.0.0.dev20250522
 Summary: SkyPilot: Run AI on Any Infra — Unified, Faster, Cheaper.
 Author: SkyPilot Team
 License: Apache 2.0

{skypilot_nightly-1.0.0.dev20250521.dist-info → skypilot_nightly-1.0.0.dev20250522.dist-info}/RECORD

@@ -1,20 +1,20 @@
-sky/__init__.py,sha256=FdlnKwsN40b-2AZDV73OlqLnhQS6OTjJp8zfjDdezyo,6428
+sky/__init__.py,sha256=edBlvvmnGPlfqmHfDb3djjLpdBo43JGlFT5B2nOlEjY,6428
 sky/admin_policy.py,sha256=hPo02f_A32gCqhUueF0QYy1fMSSKqRwYEg_9FxScN_s,3248
 sky/authentication.py,sha256=ND011K_-Ud1dVZF37A9KrwYir_ihJXcHc7iDWmuBc8Q,22872
 sky/check.py,sha256=I3vs6IhgpG3iDsOM1U3IFfYhCpELTn_mY-fyrGo45Mo,16549
 sky/cli.py,sha256=nVqxLz3c-5xZi5162rCDsjYGrVwFXb5cOpFiPE_0O6Y,235902
 sky/cloud_stores.py,sha256=Ln5GBpel-sEs7rVx7bBrMkfLwA_bctI05Rox2uoz7Lo,26388
-sky/core.py,sha256=AT-ZX3NXsIyRBMzdtEBlhc1bAj6CcufiuAGsjT4MX94,49469
+sky/core.py,sha256=zfViBwx4jOj2ydH99h9h0w55WGt1ssR0VkjPOBAVpVQ,49467
 sky/dag.py,sha256=8x-VMtjvSi0lYBemCMPLYq5ONljhoABjWzMKjmmdjSo,3369
 sky/exceptions.py,sha256=6TJ1MMw6Y9W94eig22xgun3rzvJCTh4yHNq3mGACX-U,17362
-sky/execution.py,sha256=nRevwm-z6A9UvcN1GzQA7etSBp5Hv6JL1oT1HxB5Sp8,31325
+sky/execution.py,sha256=YSzpSxoUqLlGZ5v-2pgCt9Jkkcwz12FovOeFs3lC3q8,33109
 sky/global_user_state.py,sha256=PuAcfsmVAn5NnDKJWIasg93tO0etUfrvwjU531LhP_0,33902
 sky/models.py,sha256=bGMSATMkSMr_Kp6SCoiJVVeebwSdZuzjw_jrJzVWAAc,1603
-sky/optimizer.py,sha256=2svch0EzxT3a2AZ9xld9lLXH5MUjIyWt_SRJhkF9--s,60556
-sky/resources.py,sha256=VQQ4BctTFm-N7R4aQnlYvbZbjPYTl7_nJrAJyUjVKtE,83759
+sky/optimizer.py,sha256=qdzEjj3tkpr1M2mbV-zTyGAWqc2rt8tFr4_y2MfEkLA,60496
+sky/resources.py,sha256=1E-a81wqEDoC3kX2E5S2tHYaGKY9ST0liVR_M0bnKAg,92090
 sky/sky_logging.py,sha256=cMurxhFExKEFX1frcMR71Ti_s9Obg9WY30veVxsZB6o,7285
-sky/skypilot_config.py,sha256=kQUpx0QwMM0NiQO9qoMhsALDlJCPCQKchqZ1WyjXW8c,21188
-sky/task.py,sha256=FNpcg5Ty5cCQEv9hlFgnFbSiSC4LVgG1ejX_k3P5QjA,57190
+sky/skypilot_config.py,sha256=IYqynjPivI673up28i-lp5uTsOAzM_eDD-RGKJDilkw,24497
+sky/task.py,sha256=trVBbnDejpRjDjpl-9FTt32sx6HXihAkCmQ4yKFebXg,58403
 sky/adaptors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sky/adaptors/aws.py,sha256=4caUTO5nxZQyDVPyQdoPljaF-Lz_Fa6NEnu3FfmLZd4,8633
 sky/adaptors/azure.py,sha256=7l5jobSTsTUcTo3ptrgOpRgngHY92U64eQBPxvDe1HA,21986
@@ -45,15 +45,15 @@ sky/benchmark/benchmark_utils.py,sha256=7rf-iHt6RXZ_pnBBWOMwcdodHQW69x27xNyx0yVo
 sky/client/__init__.py,sha256=pz6xvVSd9X-gwqbsDL0E9QOojYqM0KAD0j-NCyCIF1k,38
 sky/client/cli.py,sha256=nVqxLz3c-5xZi5162rCDsjYGrVwFXb5cOpFiPE_0O6Y,235902
 sky/client/common.py,sha256=E_5cjxd8fWRB7fU1yfIbiyQf-IyVhpD5KkB7Fl3cQEI,15215
-sky/client/sdk.py,sha256=-A6gRAtPXBX8kE2DnRoac544Xru-4StDG-AlMMiLuz0,72207
-sky/clouds/__init__.py,sha256=OW6mJ-9hpJSBORCgt2LippLQEYZHNfnBW1mooRNNvxo,1416
+sky/client/sdk.py,sha256=tEserGXmy_SKI231P_0id-AFbUs83v66roOqQt4tc6k,75120
+sky/clouds/__init__.py,sha256=ridBddmSX5lEt9B_cSciYs685Sx_ieCsIU-23Yq3H90,1474
 sky/clouds/aws.py,sha256=cWcyeuLbUg7ARBcy95puzoEsrxoA2NuFRYA7bzVwgA0,54983
 sky/clouds/azure.py,sha256=Q5r2wI67nj7EiW_3mKqDaRmkIHjewfaw5wU7v7gxfi0,32671
-sky/clouds/cloud.py,sha256=3s8kqUqvM4Ajyfy8AAZbEFAoiKx7YUe0uUXmwo5wi-4,37135
+sky/clouds/cloud.py,sha256=IFDoRodW5AK9u64PuJ4SC39TbyK3UfInFkuHXvuLGiQ,37259
 sky/clouds/cudo.py,sha256=sZ77kQ_7d2XYDj0FdFqe7ygSOUVRpj35_M_IOuOFVU8,13423
 sky/clouds/do.py,sha256=r4d5FDNMQ_HwRWxqTI1KCfRX1tP2MTUwvISqlJLKqM4,11803
 sky/clouds/fluidstack.py,sha256=KxkbwXTluliDtQGjx7hxUTs6VYX2mc7UE9vJN8r8xiQ,12882
-sky/clouds/gcp.py,sha256=YVOf9L3qj8xMMBNmqefeKQKO0N_Q93uf_8PKe-Ge_J4,60498
+sky/clouds/gcp.py,sha256=5At6u9tL_-Zy-9JEbuPmwq-hVYYKa8-759aefcQlWLg,66655
 sky/clouds/ibm.py,sha256=jNJvnBL93aBu0H6jU5308vKFyplfxaNE4HUtaktRxBs,22280
 sky/clouds/kubernetes.py,sha256=oK_bLsbCYsLzDTx4vQa6Ovmf4DASjIpk_J5rPe_Y9c0,38267
 sky/clouds/lambda_cloud.py,sha256=NPs7hFrKkdPswNZoOIxXwd_Ee0LBhG3s8zi0fWKiEiU,13001
@@ -64,12 +64,12 @@ sky/clouds/runpod.py,sha256=vR3QRKRvekLMmwb9RJh_RxSlpPh1o8XfuRCnJm-zaB0,12733
 sky/clouds/scp.py,sha256=DrBluM6N-glxyI2oK05Ff-Kkz3XVC-UvcZZ4y5TUUJk,16226
 sky/clouds/vast.py,sha256=XezLilfoULFFTwZqVuLx1nPXVfTlg9S65yZS_VAG3KE,11548
 sky/clouds/vsphere.py,sha256=-MBwC8Ih4LL_jiXSrGPq4e7p8M6bq4IYZVCHnxQVJzU,12655
-sky/clouds/service_catalog/__init__.py,sha256=WCYMAhek36Ee1HjJL9LP287a70hv2NMA_8plcZDI4Z4,14997
+sky/clouds/service_catalog/__init__.py,sha256=rw8ynYIQ3RyC9HP47J3NGhMg5rK4klYHfe1ABPVGcy4,15126
 sky/clouds/service_catalog/aws_catalog.py,sha256=PbYD37rU_8m-Y_5xTglW21ppxI0GecM1sdO1yXuPwHE,13518
 sky/clouds/service_catalog/azure_catalog.py,sha256=5Q51x_WEKvQ2YSgJvZHRH3URlbwIstYuwpjaWW_wJlw,8149
-sky/clouds/service_catalog/common.py,sha256=6TCUE1kg2FvUZXpV2ktrnsRchUYNTGx8rq16m5Nztek,27766
+sky/clouds/service_catalog/common.py,sha256=1Mr0VDxiHyBaIlS7VqTy_3p7Xc0DcUFbrq1ICXsiE-Q,28307
 sky/clouds/service_catalog/config.py,sha256=ylzqewdEBjDg4awvFek6ldYmFrnvD2bVGLZuLPvEVYA,1793
-sky/clouds/service_catalog/constants.py,sha256=N-Qgy93rtWC4lMNO7eeYYgXH9LXn-Y3lcWb8lR3vMwk,435
+sky/clouds/service_catalog/constants.py,sha256=O_PahpFfvW6lWXTIrJz40OmL6dEU1naEMeRR2jEsY5s,566
 sky/clouds/service_catalog/cudo_catalog.py,sha256=V_takvL6dWTGQaTLCEvjKIotCDPnMujiNUZ87kZKGVI,4673
 sky/clouds/service_catalog/do_catalog.py,sha256=Cug2QaQlSN6nFhba7f1ksyzs6z0ICTj6vSiR-792WnI,3698
 sky/clouds/service_catalog/fluidstack_catalog.py,sha256=21-cvrYEYTIi7n3ZNF2e7_0QX-PF4BkhlVJUWQOvKrY,5059
@@ -101,13 +101,15 @@ sky/clouds/utils/azure_utils.py,sha256=NToRBnhEyuUvb-nBnsKTxjhOBRkMcrelL8LK4w6s4
 sky/clouds/utils/gcp_utils.py,sha256=YtuS4EoAMvcRnGPgE_WLENPOPWIdvhp7dLceTw_zfas,7114
 sky/clouds/utils/oci_utils.py,sha256=0YxhgZdeIHQUI1AZ86YuswsZg5HdVCIVfSTRJsSHYI0,6396
 sky/clouds/utils/scp_utils.py,sha256=MqawUhhFHHxVnn29nOI4gJ_nF665ich4Po7bsy1afsA,15948
-sky/dashboard/out/404.html,sha256=ZB7QPBMgUfhofFdSHcRNsdED5LHVknn9cui95Pz3jFE,2296
-sky/dashboard/out/clusters.html,sha256=m8LiL0yIPYQWDTC_Vqid_coMGuhLmVoUoFitiNb1qVc,12642
+sky/dashboard/out/404.html,sha256=ers0b40CeuiOUYcIzBzNkXAYa1XnjFfSBkNml3G7VtE,2296
+sky/dashboard/out/clusters.html,sha256=gt-DtYKzqvUickV1UzKrsR2DBdLhSdxGCq8BrzUBao0,12642
 sky/dashboard/out/favicon.ico,sha256=XilUZZglAl_1zRsg85QsbQgmQAzGPQjcUIJ-A3AzYn8,93590
-sky/dashboard/out/index.html,sha256=pytiY9y4lyo9VHWQXXucutdZHfqwqycG17kx94OPQQk,1407
-sky/dashboard/out/infra.html,sha256=gowJTm_2wqu2Pf4BEznS7TYLNUsYQ6qxA3nIeCuAT_E,16725
-sky/dashboard/out/jobs.html,sha256=LdU_W6DqqMwfiRFjnn0Ic8VgJQCqGUrdJwtY-0lx_Yo,13738
+sky/dashboard/out/index.html,sha256=KfEo7kiejnbVpt82Fcmxj68NNegbqU1n4vXGjL-n6fs,1407
+sky/dashboard/out/infra.html,sha256=dTVMnhTKGGQNfyOWd6fsH9MNRGN0qXXEXMaOB7hH7PQ,16725
+sky/dashboard/out/jobs.html,sha256=Z10DoLW3FWH8jyzcYBqE6GUqEWBPgjE-YdLOjPCuPfo,13738
 sky/dashboard/out/skypilot.svg,sha256=c0iRtlfLlaUm2p0rG9NFmo5FN0Qhf3pq5Xph-AeMPJw,5064
+sky/dashboard/out/_next/static/CzOVV6JpRQBRt5GhZuhyK/_buildManifest.js,sha256=Z7ajzdfoGlasZgCbkheB_4MYbHMR8Y9MvRJd1lBRuuo,1120
+sky/dashboard/out/_next/static/CzOVV6JpRQBRt5GhZuhyK/_ssgManifest.js,sha256=Z49s4suAsf5y_GfnQSvm4qtq2ggxEbZPfEDTXjy6XgA,80
 sky/dashboard/out/_next/static/chunks/236-1a3a9440417720eb.js,sha256=Lc5O6gmyJI1sTk_N_MiG5uxVFn2Y48Oo65wpMp7_0wg,25620
 sky/dashboard/out/_next/static/chunks/312-c3c8845990db8ffc.js,sha256=H8yGnoxM_IYM2kU-A7mESi4aV4Ph3PxbIdnM2v5Kd3M,25150
 sky/dashboard/out/_next/static/chunks/37-d584022b0da4ac3b.js,sha256=0TCmdR39ipjM5HhPyoGqJylFdWbj0jtTKzPv-t4CNpg,8557
@@ -130,17 +132,15 @@ sky/dashboard/out/_next/static/chunks/pages/clusters/[cluster]-e2fc2dd1955e6c36.
 sky/dashboard/out/_next/static/chunks/pages/clusters/[cluster]/[job]-33bc2bec322249b1.js,sha256=z9fta7OpPucgYYz28vUAiRx2YhkwgDlVmTeoODKxdWc,13001
 sky/dashboard/out/_next/static/chunks/pages/jobs/[job]-70756c2dad850a7e.js,sha256=QodX11wjkbEfBYwPqA1TaF1eO2sbY0HMaXhnTn9m0fk,17559
 sky/dashboard/out/_next/static/css/7e7ce4ff31d3977b.css,sha256=qsnE9kcljKiExjHlJ_3L3A8gYGDND72q5D_HE6JQd-8,33240
-sky/dashboard/out/_next/static/hvWzC5E6Q4CcKzXcWbgig/_buildManifest.js,sha256=Z7ajzdfoGlasZgCbkheB_4MYbHMR8Y9MvRJd1lBRuuo,1120
-sky/dashboard/out/_next/static/hvWzC5E6Q4CcKzXcWbgig/_ssgManifest.js,sha256=Z49s4suAsf5y_GfnQSvm4qtq2ggxEbZPfEDTXjy6XgA,80
-sky/dashboard/out/clusters/[cluster].html,sha256=P3h0h9hV3tTlvknu49KNx8OPtAjbBbWMkbRA_NDoEkQ,1984
-sky/dashboard/out/clusters/[cluster]/[job].html,sha256=_7T88nqJzgwAGLUSXE3H2WL6YcD_hEB_Vk14e7gYeqw,1653
-sky/dashboard/out/jobs/[job].html,sha256=xf79-RmqZWkq7Vy3-mdkzQ4rQFIN_qVEL6SjNNMXFns,1621
+sky/dashboard/out/clusters/[cluster].html,sha256=31MVoy_I3Wxw8wjGpwkpDq8GlP1EeyT6fVU_Yv3gRJA,1984
+sky/dashboard/out/clusters/[cluster]/[job].html,sha256=E3ajH4sKDEk9N75Bss8ZCkAPDcCNGUpUmqQP-ZiZZ6E,1653
+sky/dashboard/out/jobs/[job].html,sha256=1M2Lbo242deQ5GJKBNieC_msozn78muV_WuCeiK769Q,1621
 sky/dashboard/out/videos/cursor-small.mp4,sha256=8tRdp1vjawOrXUar1cfjOc-nkaKmcwCPZx_LO0XlCvQ,203285
 sky/data/__init__.py,sha256=Nhaf1NURisXpZuwWANa2IuCyppIuc720FRwqSE2oEwY,184
 sky/data/data_transfer.py,sha256=N8b0CQebDuHieXjvEVwlYmK6DbQxUGG1RQJEyTbh3dU,12040
 sky/data/data_utils.py,sha256=CNYPM963qby5ddW0DZNbhiWXkqgB9MHh_jrC5DoBctM,33437
 sky/data/mounting_utils.py,sha256=6f1d0EeBj4dY-LQPwh8EtI6yoEHZawDgHaC8LChDS2s,21946
-sky/data/storage.py,sha256=xumCpKDW-FELA0m0IXSrRpaAS2zV162NJWfy-ugCRjc,236587
+sky/data/storage.py,sha256=HfV0zWnIXlo8jJ37RtOha4CkeujA-Xo1NvUsBgFNI_w,236609
 sky/data/storage_utils.py,sha256=u8PTKUkE7qYwr1GgAJ45pI5YUkhUaPQPRUz_CZZo_HI,13785
 sky/jobs/__init__.py,sha256=qoI53-xXE0-SOkrLWigvhgFXjk7dWE0OTqGPYIk-kmM,1458
 sky/jobs/constants.py,sha256=FLPAT02CmG7PLVPpYOUlbw1s_LgRdQiwdg2YBg8joFA,3305
@@ -155,7 +155,7 @@ sky/jobs/dashboard/dashboard.py,sha256=JKg8cCH_Y0sf3MoDTx85BghVEXWpp8ItPLshp09-_
 sky/jobs/dashboard/static/favicon.ico,sha256=uYlvgxSM7gjBmXpZ8wydvZUPAbJiiix-rc2Xe5mma9s,15086
 sky/jobs/dashboard/templates/index.html,sha256=NrlTDiEHJDt7sViwWgXUSxVCyVl_IEukE5jdvN8WhtQ,33132
 sky/jobs/server/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-sky/jobs/server/core.py,sha256=yYoEAUWLsaBcNyMqncswwgApP3neioJxXEIgfb5FsfM,26684
+sky/jobs/server/core.py,sha256=xcas0s1PIJ1Y57E109uI7N9KWuKrodLPy1qLCZsU0QI,26802
 sky/jobs/server/dashboard_utils.py,sha256=2Mbx40W1pQqPEPHsSDbHeaF0j5cgyKy-_A9Owdwp_AQ,2315
 sky/jobs/server/server.py,sha256=LqBWzYVMMJ6rRcGeDsqCQco6pG_CcDCAHzmVbt57IQM,8618
 sky/provision/__init__.py,sha256=LzOo5LjkRXwSf29dUqN14YbjzQu3liXLQcmweTeZ4dE,6457
@@ -191,10 +191,11 @@ sky/provision/fluidstack/fluidstack_utils.py,sha256=NdhQcwhIPGT21g7lQR-t6j-1zTA_
 sky/provision/fluidstack/instance.py,sha256=TCGLojd5mEuEaUQ1BnmRvXMOSSBjltyf7dhPG3OLdgQ,13787
 sky/provision/gcp/__init__.py,sha256=zlgjR2JoaGD7sStGStMRu9bJ62f-8NKEIyb-bFHBlzM,528
 sky/provision/gcp/config.py,sha256=RaNz4-MlxYE_V_hAtOeOMwr_lz2SZ98insjw0kJAi-o,41504
-sky/provision/gcp/constants.py,sha256=3650EDtLKekfv8L9j5lH6DUrlOFvZnYfuYc0EzrRwp8,10364
+sky/provision/gcp/constants.py,sha256=rquSueYWtWyUR4BG3oXK9PQZ-qAzBRJilZ6N73HAHks,15194
 sky/provision/gcp/instance.py,sha256=fHtOh-Qo1rk_oL7wZML2s6tKUyQxQu2nuIsb_E0x_rQ,25118
-sky/provision/gcp/instance_utils.py,sha256=EjfBsTqfzSA9Y83N17Is9LA1q69VJVd-ovG8b8ThRkY,71553
+sky/provision/gcp/instance_utils.py,sha256=RM5KOAk56TU8q1LNRoo9yl4VwQ3DDq7FAfTxpgdZVTc,72094
 sky/provision/gcp/mig_utils.py,sha256=oFpcFZoapHMILSE4iIm8V5bxP1RhbMHRF7cciqq8qAk,7883
+sky/provision/gcp/volume_utils.py,sha256=tTZ8QFbssPYJQlFZ48_DUUemywn248__Jr1-l429WK8,11856
 sky/provision/kubernetes/__init__.py,sha256=y6yVfii81WYG3ROxv4hiIj-ydinS5-xGxLvXnARVQoI,719
 sky/provision/kubernetes/config.py,sha256=-6ddvMro9LZ-y7K-hUPo8GdUGZ-01YyBzTAdkKblGXY,29623
 sky/provision/kubernetes/constants.py,sha256=dZCUV8FOO9Gct80sdqeubKnxeW3CGl-u5mxKeIb-B0M,411
@@ -261,19 +262,20 @@ sky/serve/spot_placer.py,sha256=auRlYZNg8uIW-lxiirxpDP-iIaJtkrhesm6OGMgwPq4,1128
 sky/serve/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sky/serve/client/sdk.py,sha256=7wJf_Ob1291Cs4yKAghZ_xWvd-Kc0IDtHiLIlphj8tQ,14673
 sky/serve/server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sky/serve/server/core.py,sha256=kjPrsln_hF0cM5nVFujQm5uTemtLIHDTOfPbl12FZzc,41178
+sky/serve/server/core.py,sha256=sS1fkQj05CUjZGnkj73m41YKJLyRo4zueEfKOs4y1L4,41070
 sky/serve/server/server.py,sha256=A9K37a0nQgZeN3eKWv62Oh2C5TSAReTZ9pHmztqlI-c,4396
 sky/server/__init__.py,sha256=MPPBqFzXz6Jv5QSk6td_IcvnfXfNErDZVcizu4MLRow,27
-sky/server/common.py,sha256=RPxnyaw_ArHkzo3VJdnCO-o8lazAUJZTMUBSySdUwgI,29018
+sky/server/common.py,sha256=JGkAkj1oexseiDRB0F5IvlhFZpmavHrX0Ovdf0rqjtU,30759
 sky/server/config.py,sha256=XWf5Kw4am6vMO5wcyWevbQAFH-dmKb7AMEgDzD083-M,8538
-sky/server/constants.py,sha256=Ha3A4X0kPh0FrdlM5u34Wi7OxhScPUwHqRi4-ye347Y,1112
-sky/server/server.py,sha256=TTv9DJaW60AbR0e4YIEWFkbYVJr1bnpjr91w5Gd9ArI,49002
+sky/server/constants.py,sha256=rJAuxkxj6UqSey9hQaBZ__MyZ-rRew9XrpUBNzRdxHI,1193
+sky/server/server.py,sha256=SQ_q64TySPopUgMTC_QXOTciideCHFLAu81n8bFj_Fo,50185
 sky/server/stream_utils.py,sha256=4JMHgtoXPpCT8JwtqyUcDQ9IdZFir9om0JaCRr8rvbQ,5849
 sky/server/uvicorn.py,sha256=4bTlvqQiykYDEIfV6Ta9KXWKZUtSfIrhDeT1F9yQuvs,3323
 sky/server/html/log.html,sha256=TSGZktua9Ysl_ysg3w60rjxAxhH61AJnsYDHdtqrjmI,6929
+sky/server/html/token_page.html,sha256=ha1BnR9ri4DCK-LLsqoznBM_UnRbzeqFpiGAXe-QOVM,5597
 sky/server/requests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sky/server/requests/event_loop.py,sha256=OhpPbuce65bbjpGRlcJa78AVnYSm08SzFKt70ypCUuQ,1211
-sky/server/requests/executor.py,sha256=EURzExLaHeRA1w2B52p4HWokfHmbJAVgYl_oV40w0jw,23628
+sky/server/requests/executor.py,sha256=9p_ECwh8fhfiou7k5cwrAYEDRjw4nigerc4kIomNgSg,23542
 sky/server/requests/payloads.py,sha256=qcvkd4p0rh_QAZz8MhmBNu8Xa4z8WLtxBOQR5ulGtuk,17064
 sky/server/requests/preconditions.py,sha256=ipxIb_3JXG6S3-ymcOdqQNb7VDvoPqADxu9ZK7-nQWc,7179
 sky/server/requests/process.py,sha256=uv6JmqdT1vR6S5j3a0CEmxz3fUoKQoZCryQsjZpZE7E,8734
@@ -321,7 +323,7 @@ sky/templates/azure-ray.yml.j2,sha256=2_WQ3x_LoraE3F12qc97DUXZvD9no0gs8jGRhvkV3N
 sky/templates/cudo-ray.yml.j2,sha256=9g0yFYUk6xPet-JNTPuVDZlVZ9oAcFPL1qaE6fyy-Xc,3797
 sky/templates/do-ray.yml.j2,sha256=7XZsM2H6RF2muHCm6xnqglnKhONXH8S3Y_mFiVxD558,4220
 sky/templates/fluidstack-ray.yml.j2,sha256=dNr9svMNp0z6-PHyQeRa6Hl3SbgqMESgBdfpvtP3aH0,3789
-sky/templates/gcp-ray.yml.j2,sha256=IgvSSBgYhvQxUO4CJo35wyVDJb2tE1sCKoan7UVA8zU,10403
+sky/templates/gcp-ray.yml.j2,sha256=2wK1j7nkLzZm2kHIbc76WZ-dk4yxObzBEEhzTEQUOuc,11789
 sky/templates/ibm-ray.yml.j2,sha256=n_EsoqQx1DUPDh7ZgCxvlEEZnROJ4rBXrFcHYVRIN2c,6907
 sky/templates/jobs-controller.yaml.j2,sha256=LbeWGkarOcrRbbdvkZv_ZfyydEcJBohKItCoAfWK6X0,2432
 sky/templates/kubernetes-ingress.yml.j2,sha256=73iDklVDWBMbItg0IexCa6_ClXPJOxw7PWz3leku4nE,1340
@@ -331,7 +333,7 @@ sky/templates/kubernetes-ray.yml.j2,sha256=G_2rwLWfBBTHwD8LSkt_W0Bz6bYCx6xPZvXaS
 sky/templates/kubernetes-ssh-jump.yml.j2,sha256=k5W5sOIMppU7dDkJMwPlqsUcb92y7L5_TVG3hkgMy8M,2747
 sky/templates/lambda-ray.yml.j2,sha256=AeFThho5hPQMncrOWQ6DhEWOsAuZtdxf1gjmp_6kPJA,4747
 sky/templates/local-ray.yml.j2,sha256=FNHeyHF6nW9nU9QLIZceUWfvrFTTcO51KqhTnYCEFaA,1185
-sky/templates/nebius-ray.yml.j2,sha256=AvZJXst-8cmGlLfLffp1Jsl6qNDRmSl4ttG0JEB8Pv4,6122
+sky/templates/nebius-ray.yml.j2,sha256=6TyE4BKZg6W6pRLKvVdVdmS0bqsTC_EQRb2i0w3EE_M,6058
 sky/templates/oci-ray.yml.j2,sha256=TvB2at81DuXSG1QV4dxmqjjMdfDTsCU5T4JH01vI5DU,4810
 sky/templates/paperspace-ray.yml.j2,sha256=GrMfpUKgeZ76FoKsnqggBrOPkwiYqUw297npPMOcMFw,4270
 sky/templates/runpod-ray.yml.j2,sha256=Ch306rKPPScu8Q6A_IJID31hRa46xnuqu_r0ziWZWG8,4604
@@ -346,7 +348,7 @@ sky/usage/constants.py,sha256=mFrTgrFIfFf4kpcl-M1VDU7_moD5_mJazUJTUDrybms,1102
 sky/usage/usage_lib.py,sha256=rInJW2kj2O1wwXUZAbeVVLhnoa7T_xBHqDhbBBrUqfI,21400
 sky/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sky/utils/accelerator_registry.py,sha256=rZniDbqqPAF-vjkrwxGwEErFSAp6puOimkRj3ppOSRY,3905
-sky/utils/admin_policy_utils.py,sha256=y_do0VH6qh163EqSuRW1uGeKvTnJhiYNrHUs77uoOcA,6013
+sky/utils/admin_policy_utils.py,sha256=1f-iwc0Z2h1dKxQxfYldkWyEuZLws4LERptRcBO-5qs,6020
 sky/utils/annotations.py,sha256=-rfacB30Sl0xkFriejGvxma3oKctGfXXLZkQPHG33eo,1626
 sky/utils/atomic.py,sha256=vrw-7XCnckF0xCx-ttamao7evPdGtVsnjaTtgMlBXIE,1280
 sky/utils/cluster_utils.py,sha256=s6DFRXktv6_gF_DnwDEXJ7CniifHp8CAPeGciRCbXgI,14432
@@ -355,8 +357,8 @@ sky/utils/command_runner.pyi,sha256=SDwvKvwaK4_-l2S3bwKSHZX8tGKU5SkbKuYRDn03o7o,
 sky/utils/common.py,sha256=P4oVXFATUYgkruHX92cN12SJBtfb8DiOOYZtbN1kvP0,1927
 sky/utils/common_utils.py,sha256=T850YbVaAk3VCenBAEY99jDphA2-B1q7AcuunEkfj9E,32712
 sky/utils/config_utils.py,sha256=v8yZcr8OvEfXmBDYcdOj3U-AH1XERY76zGJMNdqzvqc,10339
-sky/utils/context.py,sha256=Sqr28B898Rck9T3dA6lw1US8QSLe8p3AypAgGOFjB_E,8126
-sky/utils/context_utils.py,sha256=74uR78DgpL7HIdI2Nr3rJQDtV9vY7ezJTulwo5l3UoI,6157
+sky/utils/context.py,sha256=yxNlkVV9_7_0bnXJkXXe8h9t_LbFbYzPAnWIx8aTLJI,9327
+sky/utils/context_utils.py,sha256=cby-QPmnGObjIE4K7eZ_dkWZdUo7YJUmnJr5oKf_v54,6712
 sky/utils/control_master_utils.py,sha256=iD4M0onjYOdZ2RuxjwMBl4KhafHXJzuHjvqlBUnu-VE,1450
 sky/utils/controller_utils.py,sha256=65cA--9V32HTcNS-gaXZRcLswafS7DlMZ9qSbdTLkF8,53396
 sky/utils/dag_utils.py,sha256=FjxMaAT6ia4ndVg-ogXgiekLGD1GiPCAdK_c-cjIeuY,7607
@@ -367,10 +369,10 @@ sky/utils/kubernetes_enums.py,sha256=imGqHSa8O07zD_6xH1SDMM7dBU5lF5fzFFlQuQy00QM
 sky/utils/log_utils.py,sha256=0c6PiUE_XUZRc0MTKxWGPYKU7pM6QsOytclcDKVHtzo,16684
 sky/utils/message_utils.py,sha256=zi2Z7PEX6Xq_zvho-aEZe_J7UvpKOLdVDdGAcipRQPU,2662
 sky/utils/registry.py,sha256=I08nS0rvCF-xR5GEZoHEVgN1jcOeglz77h7xPpBCIjU,4179
-sky/utils/resources_utils.py,sha256=WpEiTpieZsePMKfhQ383B-1wy5iFc_PTRD0DxReC-Ss,11339
+sky/utils/resources_utils.py,sha256=oxHFiISc0Y68gCGhTw4ep6bJ6Fi21epMyeu3EN20Smc,11673
 sky/utils/rich_console_utils.py,sha256=wPvAlshaFHuMZSjiDnaK3OSBppZLBjAn-lj7AvxNBQk,553
 sky/utils/rich_utils.py,sha256=ZTXdFJp--4CJDt2ANQz0VICLPc2V7llgA6LelHdG9Mw,14188
-sky/utils/schemas.py,sha256=vcV0YCY-ZH8XKet92Uv_frLcYqH6Uh_Om-arwM4KDp8,35906
+sky/utils/schemas.py,sha256=FN0ncx-7GCicdcrvtgdD8PQizK9jQfOmENPCv1Ab4bE,37435
 sky/utils/status_lib.py,sha256=zn_MSuRYQdNKF8pnFOGQ54X_s_R7dyqWS6Q3a9zENw8,1512
 sky/utils/subprocess_utils.py,sha256=NmW22zGe9iXDXQ7XfsP4i5fve8Jq9KKiUwD7qUmw44k,16069
 sky/utils/timeline.py,sha256=ob6s3bc7nwAuSI76yLKBrSR5bzOHnOhbozz1avwoet4,4070
@@ -393,9 +395,9 @@ sky/utils/kubernetes/k8s_gpu_labeler_setup.yaml,sha256=VLKT2KKimZu1GDg_4AIlIt488
 sky/utils/kubernetes/kubernetes_deploy_utils.py,sha256=HPVgNt-wbCVPd9dpDFiA7t2mzQLpjXHJ61eiwRbEr-c,10378
 sky/utils/kubernetes/rsync_helper.sh,sha256=MT29sI5iD2QxYlXFwrN16oq0Er4TPFQVs4Z4A3U4a7Q,2483
 sky/utils/kubernetes/ssh_jump_lifecycle_manager.py,sha256=Kq1MDygF2IxFmu9FXpCxqucXLmeUrvs6OtRij6XTQbo,6554
-skypilot_nightly-1.0.0.dev20250521.dist-info/licenses/LICENSE,sha256=emRJAvE7ngL6x0RhQvlns5wJzGI3NEQ_WMjNmd9TZc4,12170
-skypilot_nightly-1.0.0.dev20250521.dist-info/METADATA,sha256=_8sNr6eaUj2T6jmLYJFCZ6ANw9a7tOZlpn7x-x2Qqwo,17985
-skypilot_nightly-1.0.0.dev20250521.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-skypilot_nightly-1.0.0.dev20250521.dist-info/entry_points.txt,sha256=StA6HYpuHj-Y61L2Ze-hK2IcLWgLZcML5gJu8cs6nU4,36
-skypilot_nightly-1.0.0.dev20250521.dist-info/top_level.txt,sha256=qA8QuiNNb6Y1OF-pCUtPEr6sLEwy2xJX06Bd_CrtrHY,4
-skypilot_nightly-1.0.0.dev20250521.dist-info/RECORD,,
+skypilot_nightly-1.0.0.dev20250522.dist-info/licenses/LICENSE,sha256=emRJAvE7ngL6x0RhQvlns5wJzGI3NEQ_WMjNmd9TZc4,12170
+skypilot_nightly-1.0.0.dev20250522.dist-info/METADATA,sha256=9lomf6MJ8oSHdrSuZ5EOBEsqyw_-QnW-S3rpwsbPfZo,17985
+skypilot_nightly-1.0.0.dev20250522.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+skypilot_nightly-1.0.0.dev20250522.dist-info/entry_points.txt,sha256=StA6HYpuHj-Y61L2Ze-hK2IcLWgLZcML5gJu8cs6nU4,36
+skypilot_nightly-1.0.0.dev20250522.dist-info/top_level.txt,sha256=qA8QuiNNb6Y1OF-pCUtPEr6sLEwy2xJX06Bd_CrtrHY,4
+skypilot_nightly-1.0.0.dev20250522.dist-info/RECORD,,