skypilot-nightly 1.0.0.dev20240530__py3-none-any.whl → 1.0.0.dev2024053101__py3-none-any.whl

sky/__init__.py

@@ -5,7 +5,7 @@ from typing import Optional
 import urllib.request
 
 # Replaced with the current commit when building the wheels.
-_SKYPILOT_COMMIT_SHA = 'f377809ae120aa717c6c8684bba368811e12bb8a'
+_SKYPILOT_COMMIT_SHA = 'e620ccc418ee69d70e580a703569d7cf0508b0ce'
 
 
 def _get_git_commit():
@@ -35,7 +35,7 @@ def _get_git_commit():
 
 
 __commit__ = _get_git_commit()
-__version__ = '1.0.0.dev20240530'
+__version__ = '1.0.0-dev2024053101'
 __root_dir__ = os.path.dirname(os.path.abspath(__file__))
 
 

sky/cli.py

@@ -772,6 +772,8 @@ def _make_task_or_dag_from_entrypoint_with_overrides(
     else:
         task = sky.Task(name='sky-cmd', run=entrypoint)
         task.set_resources({sky.Resources()})
+        # env update has been done for DAG in load_chain_dag_from_yaml for YAML.
+        task.update_envs(env)
 
     # Override.
     if workdir is not None:
@@ -787,7 +789,6 @@ def _make_task_or_dag_from_entrypoint_with_overrides(
         task.num_nodes = num_nodes
     if name is not None:
         task.name = name
-    task.update_envs(env)
     return task
 
 

sky/data/mounting_utils.py

@@ -12,7 +12,7 @@ _STAT_CACHE_CAPACITY = 4096
 _TYPE_CACHE_TTL = '5s'
 _RENAME_DIR_LIMIT = 10000
 # https://github.com/GoogleCloudPlatform/gcsfuse/releases
-GCSFUSE_VERSION = '1.3.0'
+GCSFUSE_VERSION = '2.2.0'
 
 
 def get_s3_mount_install_cmd() -> str:

sky/utils/command_runner.pyi

@@ -101,7 +101,8 @@ class CommandRunner:
               *,
               up: bool,
               log_path: str = ...,
-              stream_logs: bool = ...) -> None:
+              stream_logs: bool = ...,
+              max_retry: int = 1) -> None:
         ...
 
     @classmethod
@@ -191,5 +192,6 @@ class SSHCommandRunner(CommandRunner):
               *,
               up: bool,
               log_path: str = ...,
-              stream_logs: bool = ...) -> None:
+              stream_logs: bool = ...,
+              max_retry: int = 1) -> None:
         ...

sky/utils/dag_utils.py

@@ -70,9 +70,9 @@ def load_chain_dag_from_yaml(
     Has special handling for an initial section in YAML that contains only the
     'name' field, which is the DAG name.
 
-    'env_overrides' is in effect only when there's exactly one task. It is a
-    list of (key, value) pairs that will be used to update the task's 'envs'
-    section.
+    'env_overrides' is a list of (key, value) pairs that will be used to update
+    the task's 'envs' section. If it is a chain dag, the envs will be updated
+    for all tasks in the chain.
 
     Returns:
       A chain Dag with 1 or more tasks (an empty entrypoint would create a
@@ -90,12 +90,6 @@ def load_chain_dag_from_yaml(
         # YAML has only `name: xxx`. Still instantiate a task.
         configs = [{'name': dag_name}]
 
-    if len(configs) > 1:
-        # TODO(zongheng): in a chain DAG of N tasks, cli.py currently makes the
-        # decision to not apply overrides. Here we maintain this behavior. We
-        # can listen to user feedback to change this.
-        env_overrides = None
-
     current_task = None
     with dag_lib.Dag() as dag:
         for task_config in configs:

sky/utils/schemas.py

@@ -141,6 +141,8 @@ def _get_single_resources_schema():
                 }, {
                     'type': 'object',
                     'required': [],
+                }, {
+                    'type': 'null',
                 }]
             },
             # The following fields are for internal use only.

{skypilot_nightly-1.0.0.dev20240530.dist-info → skypilot_nightly-1.0.0.dev2024053101.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: skypilot-nightly
-Version: 1.0.0.dev20240530
+Version: 1.0.0.dev2024053101
 Summary: SkyPilot: An intercloud broker for the clouds
 Author: SkyPilot Team
 License: Apache 2.0

{skypilot_nightly-1.0.0.dev20240530.dist-info → skypilot_nightly-1.0.0.dev2024053101.dist-info}/RECORD

@@ -1,7 +1,7 @@
-sky/__init__.py,sha256=u17kWPRuqFEP-jnBDW3_jqsnpmvVEcSltMgHdqisF-0,5588
+sky/__init__.py,sha256=JsZC-jmxDHFRuZWryJ5Ktt9MWnJo2RqMdktIzikC5Ws,5590
 sky/authentication.py,sha256=LM8vOdxE7KkZSdkCw6Pi9tnCobHk8qnX2Mggi-rSSe8,21408
 sky/check.py,sha256=COsYRQbsJ78QGYlSdiqeYjK1nsipBvBcotqJlYhuQ68,9079
-sky/cli.py,sha256=-auY1_yTFJxPg8Y3DREmtQxwwf4crly7SWieWrkUXkc,200458
+sky/cli.py,sha256=H2Rcsb3ZqRAz7QoeYrC8adDiLNoYGMSpdXt26UnK4m4,200543
 sky/cloud_stores.py,sha256=KEt1EN6RtF7j5-QqmMAf2KYVX65APAdfPWY_6CVgBOg,11806
 sky/core.py,sha256=VbTyO2AEEP-00nf-JQejWSBlVyq-SdYF9hoZ33jlDXY,34233
 sky/dag.py,sha256=d3gF030wYmit01jxszEygT4EvKJUky3DBlG3PfWPp_w,2529
@@ -88,7 +88,7 @@ sky/clouds/utils/scp_utils.py,sha256=nvChcW5XS3BvMNRHtTgQ13nK3Ebh_7Fqq5J9o3g7fyM
 sky/data/__init__.py,sha256=Nhaf1NURisXpZuwWANa2IuCyppIuc720FRwqSE2oEwY,184
 sky/data/data_transfer.py,sha256=MBmjey9_p2L3IKNKTi8um09SlZe32n4wK3CkVnlTVvo,7346
 sky/data/data_utils.py,sha256=e1fXnoaz9RzfEKcXeF_5XBDwKKMYv--prE34xMwVTPo,21664
-sky/data/mounting_utils.py,sha256=4qkfMd0afvp2bEPxiM_DCZ3uxJZ-ztEPBpOplGBm3wA,8351
+sky/data/mounting_utils.py,sha256=b_70EdB4EQC7gTlIM5xDpJHx34T-l0-DNrkDnZf66JI,8351
 sky/data/storage.py,sha256=p14z1waNAhK7tGqnUG4i_Xj8I47O6u2J85AH1ICgJ0g,118872
 sky/data/storage_utils.py,sha256=-s0iQhV8JVx1J2gWtoBFrN04MGv2oVYxo_Hw43R2BSY,6867
 sky/jobs/__init__.py,sha256=9cqFutVlfjQb7t8hzG-ZlQmMlbmfMirn0KNBxIFnJYQ,1398
@@ -241,17 +241,17 @@ sky/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sky/utils/accelerator_registry.py,sha256=BO4iYH5bV80Xyp4EPfO0n1D3LL0FvESCy7xm59Je3_o,3798
 sky/utils/cluster_yaml_utils.py,sha256=1wRRYqI1kI-eFs1pMW4r_FFjHJ0zamq6v2RRI-Gtx5E,849
 sky/utils/command_runner.py,sha256=9xQuN5w_D706NUw-NtZi96tb12odfX_uoYNGlb261vI,23317
-sky/utils/command_runner.pyi,sha256=aMRsWFezJC0__pabqPN8_DBNXP81AdJubknPI8JkeEk,5492
+sky/utils/command_runner.pyi,sha256=C3feaqF93xjEzuBH4Xpa0LHp4su8JNr3GeXOZb0NjUw,5560
 sky/utils/common_utils.py,sha256=kqYNgs1zbNJ9c7qkfWd_9w6Y3cx1ln-U7V2RZeZYDi0,23848
 sky/utils/controller_utils.py,sha256=JFhAt2E_Xe7uIn0cMfXhqWWt5EcdGvhyZ-JMFoJ351E,34620
-sky/utils/dag_utils.py,sha256=-JmGlGpQuFPRCBVSG2EabyeGKf1kSWSNDGXitgBz860,5731
+sky/utils/dag_utils.py,sha256=ddSmnOjE9yMpaSC5jlC-x_tq1hPsVYJzBTyDch_cpr4,5487
 sky/utils/db_utils.py,sha256=AOvMmBEN9cF4I7CoXihPCtus4mU2VDGjBQSVMMgzKlA,2786
 sky/utils/env_options.py,sha256=1VXyd3bhiUgGfCpmmTqM9PagRo1ILBH4-pzIxmIeE6E,861
 sky/utils/kubernetes_enums.py,sha256=imGqHSa8O07zD_6xH1SDMM7dBU5lF5fzFFlQuQy00QM,1384
 sky/utils/log_utils.py,sha256=W7FYK7xzvbq4V-8R-ihLtz939ryvtABug6O-4DFrjho,8139
 sky/utils/resources_utils.py,sha256=cugCPbi03MdiCliT2OCoxnEkpaZzVWd-IbrKVtutdaw,5540
 sky/utils/rich_utils.py,sha256=5ZVhzlFx-nhqMXwv00eO9xC4rz7ibDlfD2lmGhZrJEY,1581
-sky/utils/schemas.py,sha256=iC3UTlSS8Vj3tr1oJwYu0_ce_68Xk61H5BqA9f3Xqd0,23359
+sky/utils/schemas.py,sha256=qdkQZge83SjE2T_MsyI94tifk-75QEFIRSbLXzoVVJ0,23416
 sky/utils/subprocess_utils.py,sha256=_lUMbkr9SAkLb2GdsLUowIYH8_pwm-NqxNTRZUrAfdA,6509
 sky/utils/timeline.py,sha256=ao_nm0y52ZQILfL7Y92c3pSEFRyPm_ElORC3DrI5BwQ,3936
 sky/utils/ux_utils.py,sha256=318TRunQCyJpJXonfiJ1SVotNA-6K4F2XgMEYjvWvsk,3264
@@ -267,9 +267,9 @@ sky/utils/kubernetes/gpu_labeler.py,sha256=MEUv0U4ACDcNwtFVltlv017XJMjxx1Bndf6fL
 sky/utils/kubernetes/k8s_gpu_labeler_job.yaml,sha256=KPqp23B-zQ2SZK03jdHeF9fLTogMEFxotE35IosIpRg,1186
 sky/utils/kubernetes/k8s_gpu_labeler_setup.yaml,sha256=VLKT2KKimZu1GDg_4AIlIt488oMQvhRZWwsj9vBbPUg,3812
 sky/utils/kubernetes/ssh_jump_lifecycle_manager.py,sha256=RFLJ3k7MR5UN4SKHykQ0lV9SgXumoULpKYIAt1vh-HU,6560
-skypilot_nightly-1.0.0.dev20240530.dist-info/LICENSE,sha256=emRJAvE7ngL6x0RhQvlns5wJzGI3NEQ_WMjNmd9TZc4,12170
-skypilot_nightly-1.0.0.dev20240530.dist-info/METADATA,sha256=h5bj_bpTQhxxepkd4MP0-BnA9hm6yBVNO4EwvuuPtq4,17739
-skypilot_nightly-1.0.0.dev20240530.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-skypilot_nightly-1.0.0.dev20240530.dist-info/entry_points.txt,sha256=StA6HYpuHj-Y61L2Ze-hK2IcLWgLZcML5gJu8cs6nU4,36
-skypilot_nightly-1.0.0.dev20240530.dist-info/top_level.txt,sha256=qA8QuiNNb6Y1OF-pCUtPEr6sLEwy2xJX06Bd_CrtrHY,4
-skypilot_nightly-1.0.0.dev20240530.dist-info/RECORD,,
+skypilot_nightly-1.0.0.dev2024053101.dist-info/LICENSE,sha256=emRJAvE7ngL6x0RhQvlns5wJzGI3NEQ_WMjNmd9TZc4,12170
+skypilot_nightly-1.0.0.dev2024053101.dist-info/METADATA,sha256=7ygdKz5YBSZ0M3IGD7JUbtAp6QWA8VMogH_a-7izJXE,17741
+skypilot_nightly-1.0.0.dev2024053101.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+skypilot_nightly-1.0.0.dev2024053101.dist-info/entry_points.txt,sha256=StA6HYpuHj-Y61L2Ze-hK2IcLWgLZcML5gJu8cs6nU4,36
+skypilot_nightly-1.0.0.dev2024053101.dist-info/top_level.txt,sha256=qA8QuiNNb6Y1OF-pCUtPEr6sLEwy2xJX06Bd_CrtrHY,4
+skypilot_nightly-1.0.0.dev2024053101.dist-info/RECORD,,