lightning-sdk 0.1.55__py3-none-any.whl → 0.1.57__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_validate_deployment_image_request.py

@@ -42,21 +42,26 @@ class V1ValidateDeploymentImageRequest(object):
     """
     swagger_types = {
         'image': 'str',
+        'project_id': 'str',
         'secret': 'str'
     }
 
     attribute_map = {
         'image': 'image',
+        'project_id': 'projectId',
         'secret': 'secret'
     }
 
-    def __init__(self, image: 'str' =None, secret: 'str' =None):  # noqa: E501
+    def __init__(self, image: 'str' =None, project_id: 'str' =None, secret: 'str' =None):  # noqa: E501
         """V1ValidateDeploymentImageRequest - a model defined in Swagger"""  # noqa: E501
         self._image = None
+        self._project_id = None
         self._secret = None
         self.discriminator = None
         if image is not None:
             self.image = image
+        if project_id is not None:
+            self.project_id = project_id
         if secret is not None:
             self.secret = secret
 
@@ -81,6 +86,27 @@ class V1ValidateDeploymentImageRequest(object):
 
         self._image = image
 
+    @property
+    def project_id(self) -> 'str':
+        """Gets the project_id of this V1ValidateDeploymentImageRequest.  # noqa: E501
+
+
+        :return: The project_id of this V1ValidateDeploymentImageRequest.  # noqa: E501
+        :rtype: str
+        """
+        return self._project_id
+
+    @project_id.setter
+    def project_id(self, project_id: 'str'):
+        """Sets the project_id of this V1ValidateDeploymentImageRequest.
+
+
+        :param project_id: The project_id of this V1ValidateDeploymentImageRequest.  # noqa: E501
+        :type: str
+        """
+
+        self._project_id = project_id
+
     @property
     def secret(self) -> 'str':
         """Gets the secret of this V1ValidateDeploymentImageRequest.  # noqa: E501

lightning_sdk/machine.py

@@ -1,32 +1,64 @@
-from lightning_sdk.utils.enum import DeprecationEnum
+from dataclasses import dataclass
+from typing import ClassVar, Optional
 
 
-class Machine(DeprecationEnum):
-    """Enum holding all supported machine types for studios."""
+@dataclass(frozen=True)
+class Machine:
+    # Default Machines
+    CPU_SMALL: ClassVar["Machine"]
+    CPU: ClassVar["Machine"]
+    DATA_PREP: ClassVar["Machine"]
+    DATA_PREP_MAX: ClassVar["Machine"]
+    DATA_PREP_ULTRA: ClassVar["Machine"]
+    T4: ClassVar["Machine"]
+    T4_X_4: ClassVar["Machine"]
+    L4: ClassVar["Machine"]
+    L4_X_4: ClassVar["Machine"]
+    L4_X_8: ClassVar["Machine"]
+    A10G: ClassVar["Machine"]
+    A10G_X_4: ClassVar["Machine"]
+    A10G_X_8: ClassVar["Machine"]
+    L40S: ClassVar["Machine"]
+    L40S_X_4: ClassVar["Machine"]
+    L40S_X_8: ClassVar["Machine"]
+    A100_X_8: ClassVar["Machine"]
+    H100_X_8: ClassVar["Machine"]
+    H200_X_8: ClassVar["Machine"]
 
-    CPU_SMALL = "CPU_SMALL"
-    CPU = "CPU"
-    DATA_PREP = "DATA_PREP"
-    DATA_PREP_MAX = "DATA_PREP_MAX"
-    DATA_PREP_ULTRA = "DATA_PREP_ULTRA"
-    T4 = "T4"
-    T4_X_4 = "T4_X_4"
-    L4 = "L4"
-    L4_X_4 = "L4_X_4"
-    L4_X_8 = "L4_X_8"
-    A10G = "A10G"
-    A10G_X_4 = "A10G_X_4"
-    A10G_X_8 = "A10G_X_8"
-    L40S = "L40S"
-    L40 = "L40", "L40S"  # deprecated
-    L40S_X_4 = "L40S_X_4"
-    L40_X_4 = "L40_X_4", "L40S_X_4"  # deprecated
-    L40S_X_8 = "L40S_X_8"
-    L40_X_8 = "L40_X_8", "L40S_X_8"  # deprecated
-    A100_X_8 = "A100_X_8"
-    H100_X_8 = "H100_X_8"
-    H200_X_8 = "H200_X_8"
+    name: str
+    instance_type: str
+    cost: Optional[float] = None
+    interruptible_cost: Optional[float] = None
+    wait_time: Optional[float] = None
+    interruptible_wait_time: Optional[float] = None
 
     def __str__(self) -> str:
-        """String representation of the enum."""
-        return str(self.value)
+        """String representation of the Machine."""
+        return str(self.name) if self.name else str(self.instance_type)
+
+    def __eq__(self, other: object) -> bool:
+        """Machines are equal if the instance type is equal."""
+        if isinstance(other, Machine):
+            return self.instance_type == other.instance_type
+        return False
+
+
+Machine.CPU_SMALL = Machine(name="CPU_SMALL", instance_type="m3.medium")
+Machine.CPU = Machine(name="CPU", instance_type="cpu-4")
+Machine.DATA_PREP = Machine(name="DATA_PREP", instance_type="data-large")
+Machine.DATA_PREP_MAX = Machine(name="DATA_PREP_MAX", instance_type="data-max")
+Machine.DATA_PREP_ULTRA = Machine(name="DATA_PREP_ULTRA", instance_type="data-ultra")
+Machine.T4 = Machine(name="T4", instance_type="g4dn.2xlarge")
+Machine.T4_X_4 = Machine(name="T4_X_4", instance_type="g4dn.12xlarge")
+Machine.L4 = Machine(name="L4", instance_type="g6.4xlarge")
+Machine.L4_X_4 = Machine(name="L4_X_4", instance_type="g6.12xlarge")
+Machine.L4_X_8 = Machine(name="L4_X_8", instance_type="g6.48xlarge")
+Machine.A10G = Machine(name="A10G", instance_type="g5.8xlarge")
+Machine.A10G_X_4 = Machine(name="A10G_X_4", instance_type="g5.12xlarge")
+Machine.A10G_X_8 = Machine(name="A10G_X_8", instance_type="g5.48xlarge")
+Machine.L40S = Machine(name="L40S", instance_type="g6e.4xlarge")
+Machine.L40S_X_4 = Machine(name="L40S_X_4", instance_type="g6e.12xlarge")
+Machine.L40S_X_8 = Machine(name="L40S_X_8", instance_type="g6e.48xlarge")
+Machine.A100_X_8 = Machine(name="A100_X_8", instance_type="p4d.24xlarge")
+Machine.H100_X_8 = Machine(name="H100_X_8", instance_type="p5.48xlarge")
+Machine.H200_X_8 = Machine(name="H200_X_8", instance_type="p5e.48xlarge")

lightning_sdk/studio.py

@@ -155,9 +155,13 @@ class Studio:
     def cloud_account(self) -> str:
         return self._studio.cluster_id
 
-    def start(self, machine: Union[Machine, str] = Machine.CPU, interruptible: bool = False) -> None:
+    def start(self, machine: Union[Machine, str] = Machine.CPU, interruptible: Optional[bool] = None) -> None:
         """Starts a Studio on the specified machine type (default: CPU-4)."""
         status = self.status
+
+        if interruptible is None:
+            interruptible = self.teamspace.start_studions_on_interruptible
+
         if status == Status.Running:
             curr_machine = _machine_to_compute_name(self.machine) if self.machine is not None else None
             if curr_machine != _machine_to_compute_name(machine):

lightning_sdk/teamspace.py

@@ -1,9 +1,11 @@
+import os
 import warnings
 from pathlib import Path
 from typing import TYPE_CHECKING, List, Optional, Tuple, Union
 
 from lightning_sdk.agents import Agent
 from lightning_sdk.api import TeamspaceApi
+from lightning_sdk.machine import Machine
 from lightning_sdk.models import UploadedModelInfo
 from lightning_sdk.organization import Organization
 from lightning_sdk.owner import Owner
@@ -112,6 +114,10 @@ class Teamspace:
     def default_cloud_account(self) -> str:
         return self._teamspace.project_settings.preferred_cluster
 
+    @property
+    def start_studions_on_interruptible(self) -> bool:
+        return self._teamspace.project_settings.start_studio_on_spot_instance
+
     @property
     def cloud_accounts(self) -> List[str]:
         """All cloud accounts associated with that teamspace."""
@@ -176,6 +182,25 @@ class Teamspace:
 
         return tuple(mmts)
 
+    def list_machines(self, cloud_account: Optional[str] = None) -> List[Machine]:
+        if cloud_account is None:
+            cloud_account = os.getenv("LIGHTNING_CLUSTER_ID") or self.default_cloud_account
+
+        cluster_machines = self._teamspace_api.list_machines(self.id, cloud_account=cloud_account)
+        return [
+            Machine(
+                cluster_machine.instance_id,
+                cluster_machine.instance_id,
+                cost=cluster_machine.cost,
+                interruptible_cost=cluster_machine.spot_price,
+                wait_time=float(cluster_machine.available_in_seconds) if cluster_machine.available_in_seconds else None,
+                interruptible_wait_time=float(cluster_machine.available_in_seconds_spot)
+                if cluster_machine.available_in_seconds_spot
+                else None,
+            )
+            for cluster_machine in cluster_machines
+        ]
+
     def __eq__(self, other: "Teamspace") -> bool:
         """Checks whether the provided other object is equal to this one."""
         return (

{lightning_sdk-0.1.55.dist-info → lightning_sdk-0.1.57.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning_sdk
-Version: 0.1.55
+Version: 0.1.57
 Summary: SDK to develop using Lightning AI Studios
 Author-email: Lightning-AI <justus@lightning.ai>
 License: MIT License
@@ -46,6 +46,8 @@ Requires-Dist: fire
 Requires-Dist: simple-term-menu
 Requires-Dist: lightning-utilities
 Requires-Dist: docker
+Requires-Dist: wget
+Requires-Dist: click
 Provides-Extra: serve
 Requires-Dist: litserve>=0.2.5; extra == "serve"
 

{lightning_sdk-0.1.55.dist-info → lightning_sdk-0.1.57.dist-info}/RECORD

@@ -1,57 +1,59 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
-lightning_sdk/__init__.py,sha256=R8BIHDbZNTl9mpdfFcWgbugEcNjKzByOOPVJWIm18ec,970
+lightning_sdk/__init__.py,sha256=OA3km-3Y0trAWAJMgGucXNd2lF73ByuAOu0VJ59EFak,1039
 lightning_sdk/agents.py,sha256=ly6Ma1j0ZgGPFyvPvMN28JWiB9dATIstFa5XM8pMi6I,1577
-lightning_sdk/ai_hub.py,sha256=CpbpizTr3yzOCaEth0F1_mMifnR7Vf84quvthQnjzg4,6239
+lightning_sdk/ai_hub.py,sha256=dfw5CS93ehGmPlZCbbE2Pu3v0ANI6ZY05-ULbcHFqs4,7189
 lightning_sdk/constants.py,sha256=ztl1PTUBULnqTf3DyKUSJaV_O20hNtUYT6XvAYIrmIk,749
-lightning_sdk/helpers.py,sha256=odfwLvceDbSuwj2nK21o3ER7D-h1j5cppHs1S4ThT18,1835
+lightning_sdk/helpers.py,sha256=KWMWnORHItIIA3PGn71YPs-7RjzGi8IXa2kQ5Qo4U8M,2459
 lightning_sdk/lit_container.py,sha256=423XYYpzoGppv200kMbD9dB-iqP6xdHj7q5kejJLDKE,3778
-lightning_sdk/machine.py,sha256=qutfVwQJHC02c2ygB_O5wrFI-Eq9zbfA0E1OPEYcCO0,861
+lightning_sdk/machine.py,sha256=xWeVnKyV6AYWslookdyeYFFyYBYD6xNkxjSCAULLc6c,2706
 lightning_sdk/models.py,sha256=Ssb88nspTKs_bPnZSixsTEi30FaKkmUEEVlf7ElQQ6s,5610
 lightning_sdk/organization.py,sha256=WCfzdgjtvY1_A07DnxOpp74V2JR2gQwtXbIEcFDnoVU,1232
 lightning_sdk/owner.py,sha256=t5svD2it4C9pbSpVuG9WJL46CYi37JXNziwnXxhiU5U,1361
 lightning_sdk/plugin.py,sha256=qv0oUuclcimQUcw4wsgNrbwKT-kjsDq65Vn78i3X_4c,14936
 lightning_sdk/status.py,sha256=lLGAuSvXBoXQFEEsEYwdCi0RcSNatUn5OPjJVjDtoM0,386
-lightning_sdk/studio.py,sha256=cDdVM70JXUj4tAjGBg6Purrd-_g92HNe_YgnOCxcMdU,19576
-lightning_sdk/teamspace.py,sha256=j6EghDJvxrYoWT1xMOA4qGAw3ezIpEytqGr-QlFnRXE,13247
+lightning_sdk/studio.py,sha256=_YHanwAkyr9K8BuppGdq7Twulya9Heo5_Ta3huoJSII,19696
+lightning_sdk/teamspace.py,sha256=IjdPA5PWKWALUmQyl0NQ7wKdtFztEO6grIfLUUwnOZg,14365
 lightning_sdk/user.py,sha256=vdn8pZqkAZO0-LoRsBdg0TckRKtd_H3QF4gpiZcl4iY,1130
 lightning_sdk/api/__init__.py,sha256=Qn2VVRvir_gO7w4yxGLkZY-R3T7kdiTPKgQ57BhIA9k,413
 lightning_sdk/api/agents_api.py,sha256=G47TbFo9kYqnBMqdw2RW-lfS1VAUBSXDmzs6fpIEMUs,4059
-lightning_sdk/api/ai_hub_api.py,sha256=NIFZcwibvicCK-oj9Vfm9ylGyXCFRn8gHIo-nmncKgY,5904
-lightning_sdk/api/deployment_api.py,sha256=AQh84-N2afeLIXvTgqkZUpOslt_7Tioi6_GUG_J1AWI,21809
-lightning_sdk/api/job_api.py,sha256=KhSZfjNlKtRFpT0yhVFyQ1tOOB16_C2UYWk_AqWrXro,12495
+lightning_sdk/api/ai_hub_api.py,sha256=Yr9VvxueIqPUUeMqExbjbWD5CE_yeWjq0dXQQv47erg,6779
+lightning_sdk/api/deployment_api.py,sha256=SwpTh8elvCeEE-Ilf5pJc4nM_xUXBUNWV5ejH4Y5glI,21903
+lightning_sdk/api/job_api.py,sha256=pHJMs209qMfML0uO2jV-6v5QfA8bb-whMxQfmiInb5U,12404
 lightning_sdk/api/lit_container_api.py,sha256=QJ2-6jnKztjbg-Em6KW82z-_hz8XHmzEToVvic3WcLo,4231
-lightning_sdk/api/mmt_api.py,sha256=km27mnZ7CT_TYa3ofjVrPlidjRMJk32pnznHxjVf1OM,7633
+lightning_sdk/api/mmt_api.py,sha256=UXUJaHG4R-8jDWkYh6hFYr0uJSiAYiXFiWTJn4_eVHQ,7515
 lightning_sdk/api/org_api.py,sha256=Ze3z_ATVrukobujV5YdC42DKj45Vuwl7X52q_Vr-o3U,803
-lightning_sdk/api/studio_api.py,sha256=Cfsq8HFc4uUsj8hncnhnD_TLhw0cg-ryclGowj8S6Y0,26374
-lightning_sdk/api/teamspace_api.py,sha256=KYNyfx3aUYJyPeluM9iYphcIogBc--Bt3cV4IAgY7A4,11236
+lightning_sdk/api/studio_api.py,sha256=N0XbvNvFha-et94bTlQLLjj6OPTXvrpIWgnSMUXi4UI,26327
+lightning_sdk/api/teamspace_api.py,sha256=0ga9vYpYewksiKFvKrFN-wBKPdRs_XTIfVnmdLmYmuo,11544
 lightning_sdk/api/user_api.py,sha256=sL7RIjjtmZmvCZWx7BBZslhj1BeNh4Idn-RVcdmf7M0,2598
-lightning_sdk/api/utils.py,sha256=6j5bzUgfwTzUyWHlDcjG8w2rlNtqf3L7VocTzL466tU,24432
+lightning_sdk/api/utils.py,sha256=cldsVxR3wFVC0h9997xdT96zZPLqtYXd279FSBWGs7Q,23522
 lightning_sdk/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lightning_sdk/cli/ai_hub.py,sha256=8oy6TogDiWnHuLT3cv33XEW7vPqXPA0dDMds8kX3Z4g,1649
-lightning_sdk/cli/delete.py,sha256=s3wWoi5Jh-Dz-GpdWtO0b4jWq4PJzMfr9nr_oarCoe0,3737
-lightning_sdk/cli/download.py,sha256=Fw_Gqhkb4668V5GgmDmM2q36Oogbj7q3f4VWaaqbBeY,6886
-lightning_sdk/cli/entrypoint.py,sha256=2wmth_4_aPuDMPc6LtQ3P5U-qllN5ZWpdQ37d-vKqVI,2355
+lightning_sdk/cli/ai_hub.py,sha256=cdGR4-3Medu_uVJgFPXDouQVnE9krEGtGB57T2H73r8,3166
+lightning_sdk/cli/configure.py,sha256=0od6tgVSo_oNk7SYDYN00Dhs_RLfghkinG2F3ugl0P8,5004
+lightning_sdk/cli/connect.py,sha256=PlHw1hG6x69Umfxsqb34Hz6jxfcZCjpFmGjapJTlaAQ,1423
+lightning_sdk/cli/delete.py,sha256=jeGwq_XtjPYisGQsyDAHEtrVdgsg1fjyO5zz92EPefg,6179
+lightning_sdk/cli/download.py,sha256=dMcQ08vs8XLjjTFVJccfkElhpuPDmBOFInHwbMVYWKo,10047
+lightning_sdk/cli/entrypoint.py,sha256=OBjAFdCavA5YAf9BR9KTodrr8y981nmgOn-_AQI-Rws,3355
 lightning_sdk/cli/exceptions.py,sha256=QUF3OMAMZwBikvlusimSHSBjb6ywvHpfAumJBEaodSw,169
-lightning_sdk/cli/generate.py,sha256=AwS9SzBEk30qm26-0tnL_W2tEkH5NBmZW_cnxAukDck,2025
-lightning_sdk/cli/inspect.py,sha256=VeiazH-9DNn0SrQmP-ZM5JDdy8veQy5yxVe_Dj-KQWI,1349
+lightning_sdk/cli/generate.py,sha256=QsZREYYdocXPz7dQNE1cm8TYW2FOQTa6RW7QNn_hqvg,2304
+lightning_sdk/cli/inspect.py,sha256=nksZ562tdEeJDhsH41WDbySEUJ1Te2ztyGtKvaL1Q50,2850
 lightning_sdk/cli/job_and_mmt_action.py,sha256=OV_i5K3g-NnMUF8Sq6QliQdltnhGVRdtoyZqJVXu4Ss,1659
-lightning_sdk/cli/jobs_menu.py,sha256=6VqJecnyybYjdozj7MXH1gJehskJlwP8SoewQEXDz8s,2119
+lightning_sdk/cli/jobs_menu.py,sha256=_2qsOBTdIKT5NCjwrBo9-AAn3fIvdN9IcVUVHF_KGQk,2162
 lightning_sdk/cli/legacy.py,sha256=ocTVNwlsLRS5aMjbMkwFPjT3uEYvS8C40CJ0PeRRv8g,4707
-lightning_sdk/cli/list.py,sha256=oUbmR9T94OLEfsODzA5Pc4pUwrkJpFGuMoSTCwTqjCY,5404
-lightning_sdk/cli/mmts_menu.py,sha256=OAwWywYeFSdmXBLR7raZlL7tpAhZld4r71LUHvmnZdA,2175
-lightning_sdk/cli/run.py,sha256=GAT6on_lxQWSSfqav7EkkuZEsHnqMdIedE9Bx1xOkO4,14152
-lightning_sdk/cli/serve.py,sha256=TzDbaUtE5Bj-AFucnNKqk40TkIagUtoRPfq-7RFT16g,8553
-lightning_sdk/cli/start.py,sha256=KzfJKB4sDKOpkXs43zeAjBA0-u5-Atf3Td7CvdaT5o8,1776
-lightning_sdk/cli/stop.py,sha256=K-gxJBA-9-2zFqrH7ADG3WGbfCtorfYX7zHGiIfL5nc,2585
-lightning_sdk/cli/studios_menu.py,sha256=0kQGqGel8gAbpdJtjOM1a6NEat_TnIqRNprNn8QiK58,3236
-lightning_sdk/cli/switch.py,sha256=vseOPDSBeCi1TZXe1gZ5T2-9Iik0QfEA192WIe2Yn-Y,1791
-lightning_sdk/cli/teamspace_menu.py,sha256=GVTrMAqxxL3INX83alo5wybnrl6rBBKENo01ZNkWPik,3753
-lightning_sdk/cli/upload.py,sha256=cvirl8lfPiGV4THa5akeRy1KDoPx7Klj7tTrvWePwLc,11501
+lightning_sdk/cli/list.py,sha256=JtNQIoMO9QQlbW4OSIq8pnn777hreHtaqJ7Mn2nuwcc,9991
+lightning_sdk/cli/mmts_menu.py,sha256=x4te68uBoiT95YK0cMaZRmbrsdRYeWxWRl4T6DhJg0k,2218
+lightning_sdk/cli/run.py,sha256=8L7G6nhw04OUFoHzhMy4d4CBykQ55ZMxu_HuyWPq4KM,14018
+lightning_sdk/cli/serve.py,sha256=YCCamBTsTLpPnAw8GLEKLXAVCRT2HSzvT8lq5daJJ58,8527
+lightning_sdk/cli/start.py,sha256=QXu-rDPvnM8_dwGjGsmo2dSYdROlFpd1y9PTkO9ZH00,1861
+lightning_sdk/cli/stop.py,sha256=KzykI4-mVfla0yMRvSuBAxoo06hq3gG342sVuSFpBA4,2649
+lightning_sdk/cli/studios_menu.py,sha256=TA9rO6_fFHGMz0Nt4rJ6iV80X5pZE4xShrSiyXoU-oQ,4129
+lightning_sdk/cli/switch.py,sha256=K78yxn9voyVCZR46lV29ZD6A539EPRmvg6XYgb2K6mc,1876
+lightning_sdk/cli/teamspace_menu.py,sha256=C3g3spTKgtMwoK7pnooy0MBPz4AKhFjcObkvZyZ4v04,3797
+lightning_sdk/cli/upload.py,sha256=fy7Bz1TnAzxhjuYKrJQoClSA_hzXf02b0d1FxTvsv9o,11589
 lightning_sdk/deployment/__init__.py,sha256=BLu7_cVLp97TYxe6qe-J1zKUSZXAVcvCjgcA7plV2k4,497
 lightning_sdk/deployment/deployment.py,sha256=Fn8iJVimaWB0Ngz53oJ03kBUzVe5_-3oHAbvEqFTf0s,15681
 lightning_sdk/job/__init__.py,sha256=1MxjQ6rHkyUHCypSW9RuXuVMVH11WiqhIXcU2LCFMwE,64
 lightning_sdk/job/base.py,sha256=TS5KavtfBAFbLbNqqumEizY9edjO1joSmtUdcO5CThQ,17748
-lightning_sdk/job/job.py,sha256=X4MLFZCZOtEhNlHIWZESR5mPrS4z6bhw3rSNXxuIHSA,13065
+lightning_sdk/job/job.py,sha256=1Xf0ne4wwXpkb_GJgWD8mfueJZoKR8G4lC5T6OXijlw,13075
 lightning_sdk/job/v1.py,sha256=zgdZFnlyhzOBlbwdwx6Ob9YKBrzMnJflzLXy4qyb4So,9070
 lightning_sdk/job/v2.py,sha256=9O31CwIQjwm6qZSEZtFBWbZyBuCSmyFMsHCRbQvJdBU,10283
 lightning_sdk/job/work.py,sha256=Xm5byWTcOxYI8zZ3Sz6P8HxBTBAJ3S2KucndXA5KePQ,2332
@@ -62,7 +64,7 @@ lightning_sdk/lightning_cloud/login.py,sha256=02f3itTEMJ4wKpVxOfAAfWfigVeX8G51eg
 lightning_sdk/lightning_cloud/rest_client.py,sha256=iXuigyt3ZD9fYiCPAbQbV6Yzfs4eIDQAq6gsq82q3LI,6657
 lightning_sdk/lightning_cloud/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lightning_sdk/lightning_cloud/cli/__main__.py,sha256=aHrigVV0qIp74MuYTdN6uO8a9Xp6E1aOwdR8w1fG-7c,688
-lightning_sdk/lightning_cloud/openapi/__init__.py,sha256=r_IAKZr6ktYAv7gsr-vV0yehYNJJPXWz-yRtZ10MTyA,88407
+lightning_sdk/lightning_cloud/openapi/__init__.py,sha256=7zC35Z8YKccqfiypTJsxqL3K7GincLTPG4cwthsszPs,89403
 lightning_sdk/lightning_cloud/openapi/api_client.py,sha256=pUTQMNcZmH4BhpnuAXuT7wnegaxaX26bzdEWjdoLeTo,25630
 lightning_sdk/lightning_cloud/openapi/configuration.py,sha256=H4PTkRz1QAonxKHbx3LG-gDEtdQPAxMO-bVfMfbUfJg,8306
 lightning_sdk/lightning_cloud/openapi/rest.py,sha256=ZPPr6ZkBp6LtuAsiUU7D8Pz8Dt9ECbEM_26Ov74tdpw,13322
@@ -73,24 +75,24 @@ lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py,sha256=OA4dd
 lightning_sdk/lightning_cloud/openapi/api/auth_service_api.py,sha256=NtdfsFZtQAltWyj1wYg3HNk0cxT2HWbNhAqQSLvMDgM,30948
 lightning_sdk/lightning_cloud/openapi/api/billing_service_api.py,sha256=SDJ80TezTWKhE3LR67OI6dhHDHABc4BfG7FjcFX7Ba4,68129
 lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py,sha256=DqHIqMzzOUWueLidgWq7JQs63_2iWXsN8c60XRtzyCw,370286
-lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py,sha256=NpPJGyAz5y0OZp7vZ3gz5EaEmIQPfr2i6Qo0oXzI8mQ,164867
-lightning_sdk/lightning_cloud/openapi/api/data_connection_service_api.py,sha256=OcnRdDs5PgAbmCLM0bdnUzJpTWb_OadztNua7AfVngQ,50422
+lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py,sha256=6s3_aIEe1QEk4kcq3S_4TqABlf1WZSFoHD5vMfRk18M,169531
+lightning_sdk/lightning_cloud/openapi/api/data_connection_service_api.py,sha256=ODiLVuUw09zK63GS4ilwYE4XDbTAYi7lIu5USssmyro,55071
 lightning_sdk/lightning_cloud/openapi/api/dataset_service_api.py,sha256=wCrcXOMuVs5rA4PHyrVJqpInO-Vy1hulddjCJUenPJc,23508
 lightning_sdk/lightning_cloud/openapi/api/deployment_templates_service_api.py,sha256=VK4ZuwoE0pvWEfSWr2KZNoutRtjovJFH3-6NqFmQjNg,31887
 lightning_sdk/lightning_cloud/openapi/api/endpoint_service_api.py,sha256=IgERziSMhxLxK5ATpsNpNs5Ybpu577mCH7f6yBCr7uw,113191
 lightning_sdk/lightning_cloud/openapi/api/experiments_service_api.py,sha256=lYQt6nvbaqFnRsWFf-jL3Fou-A7Hx6qgrpRgHI3eARQ,9844
 lightning_sdk/lightning_cloud/openapi/api/file_system_service_api.py,sha256=IRAf6X1wvhcW3dilmjedPGhKklIGUj3DP6HyQ5QS5WI,34592
-lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py,sha256=cZktsDckrFVS9Qr-bl2xpU6JAMofo1owPhGdfMXdt8w,163850
+lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py,sha256=1jxtUUF4-vyvZbqJIe37b6orifZlBNxzxWJWCqnoe8I,174309
 lightning_sdk/lightning_cloud/openapi/api/lightningapp_instance_service_api.py,sha256=Y9_XweSozZL_a8qCZTOgqW9cwIOx7FTMMU-ZX0eAJqs,77205
 lightning_sdk/lightning_cloud/openapi/api/lightningapp_v2_service_api.py,sha256=hORPARfDsJLtJOgieRzI_OWKMF9wEhET141vl6FPEBs,12909
 lightning_sdk/lightning_cloud/openapi/api/lightningwork_service_api.py,sha256=dXKzbr4I7TDW-PFsqgiaaw4RtYrzB1X6NZB_hxgMnts,43747
-lightning_sdk/lightning_cloud/openapi/api/lit_logger_service_api.py,sha256=lpEo-LECmEMMM5EHFqFRFUtVRBmZgMhc0Zt3vZ5tmuE,76631
+lightning_sdk/lightning_cloud/openapi/api/lit_logger_service_api.py,sha256=pNi3nVD3mWuW4hO9s0B3q2k09LRMU5CpEaaCA3OuiEk,76635
 lightning_sdk/lightning_cloud/openapi/api/lit_page_service_api.py,sha256=xf48UmWpzjs3woKVfFeyyqCLN13NA_qDepIwMw2SRUg,20457
-lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py,sha256=XYODxMTQp92xXyfnRrsDp2DXmgyXXyHM3xBnXlha9RA,35252
+lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py,sha256=fKSJfYJX18AMiCHDHao43RXemSiKk0PzCbaQzZFiPPY,35504
 lightning_sdk/lightning_cloud/openapi/api/models_store_api.py,sha256=J-rIl7uWoi6-eZR_tg9J2N0FktRGgQ3TltfSs-ByG9I,104643
 lightning_sdk/lightning_cloud/openapi/api/organizations_service_api.py,sha256=8xYTFSC4yDs32CTcEoWR1ycXX8chLQWRCsKAXJlOCls,88781
 lightning_sdk/lightning_cloud/openapi/api/profiler_service_api.py,sha256=hM2vTawBBuOYZrxN1SjZ6mU5ldvB_jUXRcLb0as0F4M,28286
-lightning_sdk/lightning_cloud/openapi/api/projects_service_api.py,sha256=a0x8UcufASV-CZzmIu-37iJiwhTlzalB07uctELD48Q,85065
+lightning_sdk/lightning_cloud/openapi/api/projects_service_api.py,sha256=E7PM1ZlmuXMLNnIQ2aEc_AUi2ssSOi-eWA9IS2hbRbI,84822
 lightning_sdk/lightning_cloud/openapi/api/quest_service_api.py,sha256=IQppAcHoy8J99xdnZVDfyr-yzmFIkolMN3caF-wm7D0,16318
 lightning_sdk/lightning_cloud/openapi/api/secret_service_api.py,sha256=-oTJJ_XbfGAKle82anvK2XKSYE9OxRedCmUfJFjB9O0,41339
 lightning_sdk/lightning_cloud/openapi/api/slurm_jobs_user_service_api.py,sha256=UptEcOYxTbX48zm2PLcYP7f04fvn2yi5h-2k7L2LEP8,36562
@@ -99,10 +101,10 @@ lightning_sdk/lightning_cloud/openapi/api/ssh_public_key_service_api.py,sha256=z
 lightning_sdk/lightning_cloud/openapi/api/storage_service_api.py,sha256=k9hHxvq37Fv6gbX7-kKc1wlFeYxImDpt8VikxufMKXY,52395
 lightning_sdk/lightning_cloud/openapi/api/studio_jobs_service_api.py,sha256=VJ7VcSvdmGbPkgmQZhaOSR47olcNPMzRlmT5J5uKsng,27877
 lightning_sdk/lightning_cloud/openapi/api/user_service_api.py,sha256=MR0jvZY0I-fFEPNpcSqrN3vl9XjrEn_D7_gYhwYeMrc,66796
-lightning_sdk/lightning_cloud/openapi/models/__init__.py,sha256=zfXm5W-Tq1ZG1Va2aK5kMsSF9boGiKu7XwBvZCMjojo,83449
+lightning_sdk/lightning_cloud/openapi/models/__init__.py,sha256=r7ZV1C-OnDFzFrOGq_2boEpVIXwZs8QZ89ez3XKDx20,84445
 lightning_sdk/lightning_cloud/openapi/models/affiliatelinks_id_body.py,sha256=X087AkCafKp2g8E2MR5ghU3pxCNNcoehZJttvuVfdC8,4274
 lightning_sdk/lightning_cloud/openapi/models/agentmanagedendpoints_id_body.py,sha256=NzKKkTs2PtMSIJJYxeh3aDBj-djBrI_qATGz7n5Oqdk,9234
-lightning_sdk/lightning_cloud/openapi/models/agents_id_body.py,sha256=iQsLbzMMvWhWMShSCA-pTIqZvfjJkUyb_mVnvVqgcXg,19633
+lightning_sdk/lightning_cloud/openapi/models/agents_id_body.py,sha256=YlRK8OVzU08SUVmzu0sSSm1CYWYqhvSoj_f_yOvEs-A,23230
 lightning_sdk/lightning_cloud/openapi/models/app_id_works_body.py,sha256=Rd3C0e06etyIua75cqb5buq2X0ZocgLkK0xkPDWTXJc,4586
 lightning_sdk/lightning_cloud/openapi/models/appinstances_id_body.py,sha256=DfwYRF64_2dLYk8QKnBvdrCgJbRwCnVcsB73WDxHwEk,5168
 lightning_sdk/lightning_cloud/openapi/models/approveautojoindomain_domain_body.py,sha256=gC2puM75rrgB36ekVmgF9rnzWE4mAp6tpaGUrlhARRU,3771
@@ -132,7 +134,7 @@ lightning_sdk/lightning_cloud/openapi/models/create_checkout_session_request_wal
 lightning_sdk/lightning_cloud/openapi/models/create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs.py,sha256=1ET9_dSRif7KwIL73DpKfq814RuvogRT9fIqq4bgCvg,14212
 lightning_sdk/lightning_cloud/openapi/models/data_connection_mount_data_connection_mount_copy_status.py,sha256=ytC9VwBzQMvpOSZJ78uekfrMmwGRaTOclX5XC3a-dMs,3343
 lightning_sdk/lightning_cloud/openapi/models/datasets_id_body.py,sha256=OcNzTWEbLGF36eY-yImNCmTwY4ZMX2P8yPuHAvc6KSk,5533
-lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py,sha256=mz0j73zJTOi2iGUrv0Q0Lv13L1lNuD9kY1GUCAsY4Qo,17363
+lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py,sha256=nji1SbQNixWMzX0Lh0pdWAJYBfGJJVM-5OGSfoByKSM,18206
 lightning_sdk/lightning_cloud/openapi/models/deploymenttemplates_id_body.py,sha256=Q-5xLj7_Rt6xt5uPLGzeyQCLiwd98PpaNrY5fMr0Dog,15829
 lightning_sdk/lightning_cloud/openapi/models/endpoints_id_body.py,sha256=-Nv94Q8XoAT3tfYVhcjutpihfoNS2b2STbTHJr1gomo,12309
 lightning_sdk/lightning_cloud/openapi/models/experiment_name_variant_name_body.py,sha256=f-pBOnWAd03BreivwAA0D1P1wEBGq2jaC1nK-GtGxVA,3910
@@ -166,11 +168,13 @@ lightning_sdk/lightning_cloud/openapi/models/id_publications_body.py,sha256=ya6b
 lightning_sdk/lightning_cloud/openapi/models/id_publications_body1.py,sha256=H-p5V2Ee0ae6meOGmv-Gm7hKeVFf4f5IREyFEumEMjc,3670
 lightning_sdk/lightning_cloud/openapi/models/id_release_body.py,sha256=j7p1BUSf74WKdWfkQv8ABwYx8egJVKf8wxxvC1R2KxY,3690
 lightning_sdk/lightning_cloud/openapi/models/id_reportlogsactivity_body.py,sha256=l4w-s7Q3VX-FKfSkUbDNA0eTmLPeXk_-k3PrzRAhviA,3830
+lightning_sdk/lightning_cloud/openapi/models/id_reportrestarttimings_body.py,sha256=-_K7RpBgPzRy5LlpPQ78avJo3KEHJ5IXM18Hcrka_pE,3869
 lightning_sdk/lightning_cloud/openapi/models/id_start_body.py,sha256=qySSqwn-O3pUA5kGoGGcr5YADSWUhY9P_EoXtgbltGw,4602
 lightning_sdk/lightning_cloud/openapi/models/id_storage_body.py,sha256=am89naxDX9PxhWQk9hLkuFlVb0WZp9UXmi0hSj5OP9I,4920
 lightning_sdk/lightning_cloud/openapi/models/id_uploads_body.py,sha256=4fKv6N6q5me42n6x9CXnVPlYjQ9uUj9N-kOSe71cXQw,3651
 lightning_sdk/lightning_cloud/openapi/models/id_uploads_body1.py,sha256=NqIds1ra2paAdJpjdMG4D4_z7RNx-omRpXdBTx6_ukc,5232
 lightning_sdk/lightning_cloud/openapi/models/id_visibility_body.py,sha256=rONXWmM-TvpznZrdEAm8Kzcnl5nk1N2uFs8RFzjm6yo,3823
+lightning_sdk/lightning_cloud/openapi/models/id_visibility_body1.py,sha256=RsBXLZhLwYV8tkU-o2zst7aXYhlJZyCyzY-e4IN7Q8Y,4530
 lightning_sdk/lightning_cloud/openapi/models/jobs_id_body.py,sha256=ogFJ-q4pH_r1u82ywp2koVLyDuFe2I8AX0hQN67SQOs,7715
 lightning_sdk/lightning_cloud/openapi/models/jobs_id_body1.py,sha256=Z8JsEoe1XLXVIH8H5ZhH4YtAFMzpbMiKXAqcLPW0McI,4347
 lightning_sdk/lightning_cloud/openapi/models/jobs_id_body2.py,sha256=ESSqHkNRBx9llPrCFF71u6OgSjBqRoxvmecpyKzN-lw,3571
@@ -184,7 +188,7 @@ lightning_sdk/lightning_cloud/openapi/models/metricsstream_create_body.py,sha256
 lightning_sdk/lightning_cloud/openapi/models/metricsstream_delete_body.py,sha256=9Ftuk903BxyRqdByfAfpGlSyhftspbw8qnRkvY3X9fU,3687
 lightning_sdk/lightning_cloud/openapi/models/metricsstream_id_body.py,sha256=gSojEmMRu66kSpE5bKs8_p882N9BiiRVAc26sad_qmw,5287
 lightning_sdk/lightning_cloud/openapi/models/model_id_versions_body.py,sha256=TOkX8A0NaecNHAqmp90pp638LI9m2zhAZhpkEglozEQ,4439
-lightning_sdk/lightning_cloud/openapi/models/model_id_visibility_body.py,sha256=lCGbi9c1wFFCc23aTwxwbKjJSls7RQJg1tOaFg-2B5o,3873
+lightning_sdk/lightning_cloud/openapi/models/model_id_visibility_body.py,sha256=ibHpPVtPyJdXVP8EjaJZiu2JEZUCTDQsVAGyfoFSfGo,4586
 lightning_sdk/lightning_cloud/openapi/models/models_model_id_body.py,sha256=4LgGgZYBg6Iabem0LrhTtS3nzNo6L6Ghanj-dQyvvlA,4378
 lightning_sdk/lightning_cloud/openapi/models/multimachinejobs_id_body.py,sha256=Xjk_5CETlmOKvvRXDwGckxqY7UPk-wQLBZqsS-ql8PY,3954
 lightning_sdk/lightning_cloud/openapi/models/multipartuploads_upload_id_body.py,sha256=i7Vm5bH1oNEVn7TgM0puUcggbW2xt-qDdinOkr-qr8I,3777
@@ -203,7 +207,7 @@ lightning_sdk/lightning_cloud/openapi/models/project_id_fileendpoints_body.py,sh
 lightning_sdk/lightning_cloud/openapi/models/project_id_getapp_body.py,sha256=nAAb7yoON2P0QrErVmz8z6pk6v3QcJYt8eWJxQXibVQ,7200
 lightning_sdk/lightning_cloud/openapi/models/project_id_invite_body.py,sha256=rOwoxuwvCWzW99g_FSw06z0kK5ptLjaVz0axEBtgXkQ,5254
 lightning_sdk/lightning_cloud/openapi/models/project_id_jobs_body.py,sha256=IGZaQYimwiS2j1h-rt4qQCnARTwN3Fv06ouUjMZ0BfY,5011
-lightning_sdk/lightning_cloud/openapi/models/project_id_litregistry_body.py,sha256=1jyi_uDg1tmGlW5ckFGWlerAMrPjKKL7jdJlkw0GzHY,3998
+lightning_sdk/lightning_cloud/openapi/models/project_id_litregistry_body.py,sha256=nOdgOb4MOvULNH6TEq4hSufDJLAqNtHHM3XqKgwI6SM,4246
 lightning_sdk/lightning_cloud/openapi/models/project_id_memberships_body.py,sha256=h9ROMHHPwk2zScssMD93JIDHkurxhzbz3hyg627H_S4,5946
 lightning_sdk/lightning_cloud/openapi/models/project_id_models_body.py,sha256=u5msnl06gwYwCBOpwJ92EsbyO4WPaAHitLEG88inqlE,6663
 lightning_sdk/lightning_cloud/openapi/models/project_id_multimachinejobs_body.py,sha256=98YWVtnWAZfx8ydzWjmhm2qGOIcasXrNOO7LGRxZf0o,7029
@@ -223,6 +227,7 @@ lightning_sdk/lightning_cloud/openapi/models/servers_server_id_body.py,sha256=0B
 lightning_sdk/lightning_cloud/openapi/models/service_artifact_artifact_kind.py,sha256=mN25FLynNyhkY5En9k53GX2n-6itjPY84xXi3xL6CsQ,3151
 lightning_sdk/lightning_cloud/openapi/models/service_health_service_status.py,sha256=AvLUZ5iHK9ZuqMbJ6O0N-sMSRfryKDU8djk4xaK9g5A,3196
 lightning_sdk/lightning_cloud/openapi/models/serviceexecution_id_body.py,sha256=_-jU6n4JExbtPpHQwqiLc3iXeSQ7alk-GKUJiVK2dqs,10757
+lightning_sdk/lightning_cloud/openapi/models/setup.py,sha256=bx7FE6bv7ujyGTyAdx84fI0o4aNSA4Gll6n94A1hIMw,4315
 lightning_sdk/lightning_cloud/openapi/models/slurm_jobs_body.py,sha256=bhq4Txf2UALD-AabsJpoxgLWXkj8InzgtySKb2hCgOg,9334
 lightning_sdk/lightning_cloud/openapi/models/snowflake_export_body.py,sha256=Avax7cvfkHiSez0JtV0Q2sxh8h9bf_xc4KicaguIP9M,8981
 lightning_sdk/lightning_cloud/openapi/models/snowflake_query_body.py,sha256=6_DTSiTjRxxroBaC6gWJbztCVdVF2nYMOcT6lIxZYsI,5249
@@ -254,7 +259,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_append_logger_metrics_response.p
 lightning_sdk/lightning_cloud/openapi/models/v1_approve_auto_join_domain_response.py,sha256=9GbCIRQB46PhDZwFwthkTjL1VsgFpWrKGm6uOEKwLrA,3797
 lightning_sdk/lightning_cloud/openapi/models/v1_artifact.py,sha256=mjM2igfo6OM3lP7sYTSd8xRr-sxxwFuZD3fWDAuwojo,6460
 lightning_sdk/lightning_cloud/openapi/models/v1_assign_variant_response.py,sha256=Fh-qSTJAnkchu9r7vOHG0hUsEOIHXKXp1Upxxs1Umps,3028
-lightning_sdk/lightning_cloud/openapi/models/v1_assistant.py,sha256=d_Ro17TBnaopiTGrTFCcukm_y_eJnN73b8Mtzp_TUcE,20816
+lightning_sdk/lightning_cloud/openapi/models/v1_assistant.py,sha256=fwjm-dKinvbSiU5McJfKWzT0l35zyKe9CeyIlnHBj-0,24397
 lightning_sdk/lightning_cloud/openapi/models/v1_assistant_knowledge_item_status.py,sha256=0Sr60a89lsbZA79_gVv6sL7Tb_dng7zSLo6aeMRGfE4,7464
 lightning_sdk/lightning_cloud/openapi/models/v1_assistant_knowledge_status.py,sha256=RL0KvWqQiy8rYDJmZIFhGSMRHVvqpQppMt6I3t8TRW0,3919
 lightning_sdk/lightning_cloud/openapi/models/v1_assistant_model_status.py,sha256=nYIG5rNikWibOLNUveaGJ-t9f-GKBv1eQtn6UObB_v8,3273
@@ -302,7 +307,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_session.py,sha256=rI
 lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_state.py,sha256=yLpcNIGMVO5LGk10PenG4uGNxee34tecmSkP7dw_JfY,3239
 lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_version.py,sha256=ZQSAgV4qEeByenlyFqX_kZBy51_yVPVqNGhrCUvZ3_g,20082
 lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_version_publication.py,sha256=n1oHRq_UjYSP_kY2lmMjtqhHUu0H-wHQ5XiDJUEvB6s,6386
-lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py,sha256=6eEEQ3FnldH8tcSwiqvOkDLbER6LMuCcOOuk79_kO3g,37672
+lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py,sha256=ItHbbCUuP-h4czrTMbsKgGxDt7y08fxlchNWs39jmeM,38543
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_availability.py,sha256=WrS0UpcTqqCu79ny5yOBJ16RNs5B4AQSXbniWE7iez0,7103
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py,sha256=22jd_N2SYqN_ZZlBltNGdS6jp9zfkIC33qN_IQHY2yg,10949
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_deletion_options.py,sha256=GyhAXDpVtfXfaUybBE0aRRvJ6d3DfD7VV3xH_fgQOlE,4859
@@ -421,7 +426,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_delete_studio_job_response.py,sh
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_user_slurm_job_response.py,sha256=kmy5RJKy63bRzBcWdMk00-vT3m2lVs05cic7jfU1JyQ,3058
 lightning_sdk/lightning_cloud/openapi/models/v1_dependency_cache_state.py,sha256=VL2MY_XvHWAd8toG-myqZlhuoEwjAIX1cemNSXxqk-k,3210
 lightning_sdk/lightning_cloud/openapi/models/v1_dependency_file_info.py,sha256=Fyh7eRcxFJ8Y4LHA3uJe_D5F9RhOxT4ChllHph_ryfQ,4571
-lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py,sha256=uGO8_1XLqnIkwmaBG5QLbc35Rl8vGwCEK0o4obWiOoI,18234
+lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py,sha256=vLSTst7paNv9iySPhVxLeAHzMEZj3kOZw1ITrFeEXok,19057
 lightning_sdk/lightning_cloud/openapi/models/v1_deployment_api.py,sha256=hfvtI_8dSwM76baVcMwNC8DOxZlmdixASKW83--83Cg,8434
 lightning_sdk/lightning_cloud/openapi/models/v1_deployment_event.py,sha256=XDpq8DM8sN2UKCIlBWGEk-vzjGQGNoc0KjyRcCzDqZ4,9994
 lightning_sdk/lightning_cloud/openapi/models/v1_deployment_event_type.py,sha256=uuogrGRymVVSz-OcYoJ7eZyFj3POXY22EAQJkXpTVm4,3197
@@ -479,6 +484,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_folder_index_status.py,sha256=Pm
 lightning_sdk/lightning_cloud/openapi/models/v1_gallery_app.py,sha256=EAt00TanGSXQoNN8QsMGZXczzr96KOGEzlOBy0ALS9k,25152
 lightning_sdk/lightning_cloud/openapi/models/v1_gallery_component.py,sha256=fxMjrIC3hIS1T7KIMX6FKf32QPsM6YQmHR0w7mZ_1JI,24426
 lightning_sdk/lightning_cloud/openapi/models/v1_gcp_data_connection.py,sha256=09-wGP49Uejc5ebCK3aino0MJ7RhJnYi0BBGijrVXfI,3671
+lightning_sdk/lightning_cloud/openapi/models/v1_gcp_data_connection_setup.py,sha256=tmrLOnzkmrdWUP8hgDRn1avzxZ0UYKRpcsu64v9je2s,3956
 lightning_sdk/lightning_cloud/openapi/models/v1_gcs_folder_data_connection.py,sha256=u8NhGu13DLniZMKfq3trmtKrLGDcezNHC-9hV6gUU9A,3731
 lightning_sdk/lightning_cloud/openapi/models/v1_generate_ssh_key_pair_request.py,sha256=3Mgl6WmKGL-R8cT7Jb0Og_w51CnR5zUtKEYmqBjEh-s,3711
 lightning_sdk/lightning_cloud/openapi/models/v1_get_affiliate_link_response.py,sha256=LLmxfO9WisUdznVS2-u3S7rCsQgOVYLEQ19v1q_F8sQ,3773
@@ -487,6 +493,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_get_agent_job_logs_metadata_resp
 lightning_sdk/lightning_cloud/openapi/models/v1_get_artifacts_page_response.py,sha256=iC3z2-zLpKn0QB97FvIwM6-omGRgQLYNft83BuQdQLg,3879
 lightning_sdk/lightning_cloud/openapi/models/v1_get_cloud_space_instance_status_response.py,sha256=mkmpn9psHLvC22e3HtkiL5ziWDAZ0UJDPShEfFcBmbY,7049
 lightning_sdk/lightning_cloud/openapi/models/v1_get_cloud_space_size_response.py,sha256=5vc8aqIlN8SilMuc_FenQxcuTHF9f2roH-8UCuVn5OY,4828
+lightning_sdk/lightning_cloud/openapi/models/v1_get_cluster_accelerator_demand_response.py,sha256=xxRf6ry2wQM3OsUhfn9wUTUjnvJu0Srv9Dn9Mlezafo,3950
 lightning_sdk/lightning_cloud/openapi/models/v1_get_cluster_credentials_response.py,sha256=5ryHCeO6hkoK-5THe_dGuBk0EnHSD21u-5AzjGzvv5A,3851
 lightning_sdk/lightning_cloud/openapi/models/v1_get_deployment_routing_telemetry_aggregated_response.py,sha256=aKIn0uwmEPINgrrWXdXlYMCV5a8kqW8MWCLg4aHLCG8,4079
 lightning_sdk/lightning_cloud/openapi/models/v1_get_deployment_routing_telemetry_response.py,sha256=t0D7StfYVPvHH60BiPAvrnrwfiFBYeZpNFL4CkPqdhY,4168
@@ -531,14 +538,14 @@ lightning_sdk/lightning_cloud/openapi/models/v1_instance_overprovisioning_spec.p
 lightning_sdk/lightning_cloud/openapi/models/v1_interrupt_server_response.py,sha256=1VE0f_vzOSbdXygI9dMusSz8nbuzZgz--XGz7ugEgjM,3040
 lightning_sdk/lightning_cloud/openapi/models/v1_invalidate_cloud_space_instance_code_settings_response.py,sha256=ELlSdF6ykC6OO2llrmmCF0K6M7ToRzefCoYziDq-fX8,3190
 lightning_sdk/lightning_cloud/openapi/models/v1_invite_project_membership_response.py,sha256=_536O6POlwvOGkS98Rzz_N5BT4CJsgeo9ve64ojbzhs,4079
-lightning_sdk/lightning_cloud/openapi/models/v1_job.py,sha256=k-mmT7Hf2R9BQLKcLqF1l3wec3onSXBw6D9hWh3BBwA,20712
+lightning_sdk/lightning_cloud/openapi/models/v1_job.py,sha256=GI9wtjmg4q1huY0ZJSRC738Bk3SyUmaxqznW48x0o8I,21612
 lightning_sdk/lightning_cloud/openapi/models/v1_job_action.py,sha256=7h_D2__bEiu3f-tx2XgVZ7ujI8Hl2bQ5J2J8mJHUvAo,3085
 lightning_sdk/lightning_cloud/openapi/models/v1_job_file.py,sha256=zsoipn0VJcGq_9JfzJJKgQr5LHfSc6lTpT82E0f5XR0,6434
 lightning_sdk/lightning_cloud/openapi/models/v1_job_health_check_config.py,sha256=lmOqNTnUCKUycMvCqRr0tAO5hZoV9CK3cA9_tGiTUNE,8205
 lightning_sdk/lightning_cloud/openapi/models/v1_job_log_entry.py,sha256=Bhjx6rx3cJy6tarZ1zLXycZwmpZInMROecDo5ODsakg,4971
 lightning_sdk/lightning_cloud/openapi/models/v1_job_logs_page.py,sha256=uPeDC0SKPuojeSAgV-6Ei43A_nXkgxvQG1qSxd8w8To,6309
 lightning_sdk/lightning_cloud/openapi/models/v1_job_logs_response.py,sha256=9Ae7cFQTJHhI1XBsnHqNGQLFXEMeAJEw7T53jY4BMZg,4467
-lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py,sha256=icngUO9YxvvXhF9sf39Qi_hTxW1BbLIkQd4F62gdczA,21686
+lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py,sha256=QgNJl2KRe-UoJF2w3wBRiSQo-RHogm8xw9K7UHL8DM4,22355
 lightning_sdk/lightning_cloud/openapi/models/v1_job_timing.py,sha256=qnnGfuKbAU_IEWYZFPaBGTX2DFAvBvgoXllQk7yhMFk,5462
 lightning_sdk/lightning_cloud/openapi/models/v1_joinable_organization.py,sha256=IkuGZ89NWOMpe8JsxXJvkRKdcDUvV4F3akA_USvl5rs,9586
 lightning_sdk/lightning_cloud/openapi/models/v1_keep_alive_cloud_space_instance_response.py,sha256=NRrP0aqQbpd8ejQUApJ2L1vUoSx1UhNOlSh4JeQ2_es,3112
@@ -649,9 +656,9 @@ lightning_sdk/lightning_cloud/openapi/models/v1_list_studio_jobs_response.py,sha
 lightning_sdk/lightning_cloud/openapi/models/v1_list_user_slurm_jobs_response.py,sha256=iyyVxOcebdLBkJaha96qGOkpPmGrkPJ9ibIg0rxyw0U,3789
 lightning_sdk/lightning_cloud/openapi/models/v1_lit_page.py,sha256=7QIdu9A4SW48JLSixhFez7o6qjbYigI9rLvMNgjAQek,15220
 lightning_sdk/lightning_cloud/openapi/models/v1_lit_page_type.py,sha256=rJaoSTMmeOXrWpOObNCuSX25KN-iz1rSTyvwv2mYL1c,3099
-lightning_sdk/lightning_cloud/openapi/models/v1_lit_registry_artifact.py,sha256=A8fy5lbTmNO6uSMg841ZGv-xBXFS6JfWo6vkfh6Nzf8,7669
-lightning_sdk/lightning_cloud/openapi/models/v1_lit_registry_project.py,sha256=olFAa1fJLy4Aa_5KOZquumzQAqm2R2FhI_fhPl4pwJk,6956
-lightning_sdk/lightning_cloud/openapi/models/v1_lit_repository.py,sha256=rYit1rSvz2qI7_j7pWp1zbi68QBbWIywXeT5WR6vO5I,8065
+lightning_sdk/lightning_cloud/openapi/models/v1_lit_registry_artifact.py,sha256=iB9Z4-PtsaYUpkUNfCHUH_obuJhhhCXCCx5HqE4L530,8386
+lightning_sdk/lightning_cloud/openapi/models/v1_lit_registry_project.py,sha256=NpcxQ7x_2cyfi_Z0tAatEZ1hiK1m2WDVvwPVDW-h5Ss,7674
+lightning_sdk/lightning_cloud/openapi/models/v1_lit_repository.py,sha256=CPWcuqUGBnEPnlrzxPfKNCCn2Acknb05fNZT3zfb3Zc,9005
 lightning_sdk/lightning_cloud/openapi/models/v1_locked_resource.py,sha256=jz6WQa3NnJZ44MhksEMrvRCg0iVJLzCRXw9hbmsC05k,6327
 lightning_sdk/lightning_cloud/openapi/models/v1_log_page.py,sha256=OMZLuY8Kd8Yg8lOsA7j8OTjW908GkvelA-LYx2RbhRA,6205
 lightning_sdk/lightning_cloud/openapi/models/v1_logger_artifact.py,sha256=-WBlpQE2VR2WDFHgADcCqq4dw3gYGj25Plu2qA806as,6525
@@ -743,6 +750,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_refresh_response.py,sha256=29Pdm
 lightning_sdk/lightning_cloud/openapi/models/v1_region_state.py,sha256=c_6TjyTxEy_eM385G_Fg7Ks9rvjeBlJVmjyQ-OumqKA,3167
 lightning_sdk/lightning_cloud/openapi/models/v1_regional_load_balancer.py,sha256=oyBYtKR29jUmcXivUaCyB_omigKVT6soHVF07l8CBKY,4421
 lightning_sdk/lightning_cloud/openapi/models/v1_report_logs_activity_response.py,sha256=zA2m8-EYsmfap7Ob-fg_XaRc42pFFAB48_p5NRz7D9g,3058
+lightning_sdk/lightning_cloud/openapi/models/v1_report_restart_timings_response.py,sha256=qVJHTSpU_6Kdm1jLPwbqDK0tJC1eiGpSoRR3XnKCltY,3070
 lightning_sdk/lightning_cloud/openapi/models/v1_request_cluster_access_request.py,sha256=J0P8mFnD0oGn61TGJ_RH5z8IBbcYSmE6P4qzgDzLMvI,5402
 lightning_sdk/lightning_cloud/openapi/models/v1_request_cluster_access_response.py,sha256=TOCHLSYkn6Q0ptQq7KTJGhp2U4abT30tqFCVNViNqhw,3070
 lightning_sdk/lightning_cloud/openapi/models/v1_request_verification_code_response.py,sha256=h87GR0UblCyayZdoxwrEwyLTtvapQKUs_tceXHzLnZs,3877
@@ -752,6 +760,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_resources.py,sha256=GA2kSAecFOIQ
 lightning_sdk/lightning_cloud/openapi/models/v1_response_choice.py,sha256=6jPtpQ55wky_xGkM1B0lTEEZ2JTwoyaQCHp4TW9rXak,5162
 lightning_sdk/lightning_cloud/openapi/models/v1_response_choice_delta.py,sha256=9FDWQK3jCg1LOCZlhtetMW3tu7hkeuUuqRrGB3iKWbc,4348
 lightning_sdk/lightning_cloud/openapi/models/v1_restart_cloud_space_instance_response.py,sha256=BR1A_H2GUnlW7yegOTsgKgymdCFivSldigEdAqzGVbA,3100
+lightning_sdk/lightning_cloud/openapi/models/v1_restart_timing.py,sha256=AK6ZQrn2EvikopJZa2K_jWyWsH_-Qus4kIPFemof_WE,4937
 lightning_sdk/lightning_cloud/openapi/models/v1_restore_deployment_release_response.py,sha256=UkaXXGq-i5YDPzBOy5Zj1nJQZJeyBs1Z9VO5a2DYPkk,3094
 lightning_sdk/lightning_cloud/openapi/models/v1_role.py,sha256=-qebDpqvEYzWLAV9o0klUom-8zOFM6a7cJhrhrsdhhY,7554
 lightning_sdk/lightning_cloud/openapi/models/v1_rolling_update_strategy.py,sha256=Mxi5aq4dve439nGwv329293wxmNOg0ny0M04CdbR3hE,4634
@@ -772,6 +781,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_service_artifact.py,sha256=Zdn3P
 lightning_sdk/lightning_cloud/openapi/models/v1_service_execution.py,sha256=0G_mR7TWZVDP9cBC7Uv-0ECkXQTe6iiJbMwI3Du04Rg,11918
 lightning_sdk/lightning_cloud/openapi/models/v1_service_health.py,sha256=UV5CcvLyHlPMh1KxYwvcmGTt3GfwaL6i62NNWg7krns,4382
 lightning_sdk/lightning_cloud/openapi/models/v1_setup_confirmed_ssh_public_key_response.py,sha256=RD_BKmYQFe4-LiaOTvhry4LO9ZAh99-IpUeiMnSAe5g,3106
+lightning_sdk/lightning_cloud/openapi/models/v1_setup_data_connection_response.py,sha256=aojyHeXpnzwv4RjEUuJyBQ-BvC1nt93sY2UJJznDEs8,3956
 lightning_sdk/lightning_cloud/openapi/models/v1_should_start_syncing_response.py,sha256=VMFOOarb05lU61dO_rxe6MmakCAHD998YwUKwyRxELQ,3906
 lightning_sdk/lightning_cloud/openapi/models/v1_signed_url.py,sha256=Nb6sQMTv9YEcmsU55ATGwHP48bIPeGq1PxwInN0I5dI,4267
 lightning_sdk/lightning_cloud/openapi/models/v1_slurm_job.py,sha256=M6YM2z53BAcGl7cJWIaj7QLUiVTkGBKIdk2kH_JqUZU,19282
@@ -811,11 +821,12 @@ lightning_sdk/lightning_cloud/openapi/models/v1_update_cloud_space_publication_r
 lightning_sdk/lightning_cloud/openapi/models/v1_update_cluster_accelerators_request.py,sha256=Hz8tGrb6oRzTbtf3AC9QL7B2TlqSEPLzHqDfhosCiB0,4867
 lightning_sdk/lightning_cloud/openapi/models/v1_update_cluster_accelerators_response.py,sha256=Eo8HbgCNJkuIG5Ccyl7wPuOk72FhAhkaPGuAltr4tSw,3100
 lightning_sdk/lightning_cloud/openapi/models/v1_update_cluster_availability_request.py,sha256=E8htNmxp_YBjhqK3HLV42a3tdu5F07-Iv0wtEFTPRjg,6603
+lightning_sdk/lightning_cloud/openapi/models/v1_update_deployment_visibility_response.py,sha256=IFxkDJt17RkQowDTQjR3Cz4USqwpNxMKRbJc0m7U28Y,3106
 lightning_sdk/lightning_cloud/openapi/models/v1_update_index_response.py,sha256=ZZFoJkImB3Kv8aamaSr2NdL4382XnQzvy0z59EPMElg,3016
 lightning_sdk/lightning_cloud/openapi/models/v1_update_lit_page_response.py,sha256=_wp-nvU-UQarSzQW9-350OXaVslfeWhgDQ0dj8zuZ60,3786
 lightning_sdk/lightning_cloud/openapi/models/v1_update_lit_repository_response.py,sha256=901zbSzX96kC0eB8sgXIFbvnGJjRyYt8msLIilUUAww,3064
-lightning_sdk/lightning_cloud/openapi/models/v1_update_metrics_stream_visibility_response.py,sha256=NyG_fUoC0E1U4531xpW6C5DGdpkJ59mRjtPZZbBSw_k,3124
-lightning_sdk/lightning_cloud/openapi/models/v1_update_model_visibility_response.py,sha256=Dw1wFjDbpaR6441qlTj5PqJoQ3uV_rCfbUAJxgNoJBc,3076
+lightning_sdk/lightning_cloud/openapi/models/v1_update_metrics_stream_visibility_response.py,sha256=-UXm9CTv66w5ccVU3xzA6tupr0kmqYitKcP7HDYY0PY,4053
+lightning_sdk/lightning_cloud/openapi/models/v1_update_model_visibility_response.py,sha256=y9piacBOLGt-OlLcYjFEyqRzPdqaHSI1tXxZfxwQlqE,3973
 lightning_sdk/lightning_cloud/openapi/models/v1_update_project_cluster_accelerators_response.py,sha256=9NujcwHJzLFuj66q_AZVdyyZrFiy7_5pwOx89eFdS_U,3142
 lightning_sdk/lightning_cloud/openapi/models/v1_update_shared_metrics_stream_response.py,sha256=mQaPeGjhJ6EohqmF9MFqRR1NKqvznWmvzBAXgMKlW98,3100
 lightning_sdk/lightning_cloud/openapi/models/v1_update_snowflake_query_response.py,sha256=NzTD_--qCS5GsJTcnYO2UNRlaQfJyJxB2BApVNGHoEk,3070
@@ -837,14 +848,14 @@ lightning_sdk/lightning_cloud/openapi/models/v1_upstream_open_ai.py,sha256=jt1qQ
 lightning_sdk/lightning_cloud/openapi/models/v1_usage.py,sha256=RhhnH9ygScZyExg06WhvMNPPRLSe8FYkIftqF-D9NIU,13408
 lightning_sdk/lightning_cloud/openapi/models/v1_usage_details.py,sha256=U7qC698Xj5tb3D93ZskG6sDf3lTXE13UTlGeDTvtRU4,14062
 lightning_sdk/lightning_cloud/openapi/models/v1_usage_report.py,sha256=iH67BcONBSLYzcZpGpKWSOzJTCpuqYt7FU4OUs8BJ9k,6076
-lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py,sha256=17tghnLOIaeLyT0Dpmy53ys0-F1dbLAWGx5udF6wpnY,75334
+lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py,sha256=KDAwGwFeAtHt8Q5n42t2_JbvBShss2i_yLP5kuO4pMY,78175
 lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_compute_config.py,sha256=3jeJfpbBpYY2B2Ao2j2N93CMO2CnvmPqndE4Lw3ZfMA,13056
 lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_flow_compute_config.py,sha256=3WpZ-lf7xPwuYyQDMdP7Uc6-dh3vf5TaaUlcMfesfMk,5208
 lightning_sdk/lightning_cloud/openapi/models/v1_user_slurm_job_action_response.py,sha256=BdNzXH8Vsf5PHjl9Rd-TVkpAgx1YC9rf8LD0js-ba20,3058
 lightning_sdk/lightning_cloud/openapi/models/v1_validate_assistant_status_response.py,sha256=3xmSCc2Mipr2I1FsksoFg0FJVuKRxqY_OJ1tSj-jE6M,5007
 lightning_sdk/lightning_cloud/openapi/models/v1_validate_auto_join_domain_response.py,sha256=xdu6JsdlI3noKRgJR4hZix7vz0K29Im2K9MMqDMCC4I,3082
 lightning_sdk/lightning_cloud/openapi/models/v1_validate_data_connection_response.py,sha256=kRXkbFCOskylDjPqcNGeFKX9fGVO69FOCyIJoQzNi6E,7141
-lightning_sdk/lightning_cloud/openapi/models/v1_validate_deployment_image_request.py,sha256=rMIqbG5vp7rf_kZIa6QgSuUQMFweC1kAEP2BIW3gy3w,4502
+lightning_sdk/lightning_cloud/openapi/models/v1_validate_deployment_image_request.py,sha256=lzjDyMeNPxpGJwkVUYTYgcb1KLrB0uTlZyMI5rxSD98,5302
 lightning_sdk/lightning_cloud/openapi/models/v1_validate_deployment_image_response.py,sha256=PbPPG_y1kHylTWu_WUvkpj_5RJTEKwTkdjBt6JebkS4,3088
 lightning_sdk/lightning_cloud/openapi/models/v1_validate_managed_endpoint_request.py,sha256=k7tGGr12edn9NG7571mEDDj89T9PfLPoTXsHANDdLQg,5281
 lightning_sdk/lightning_cloud/openapi/models/v1_validate_managed_endpoint_response.py,sha256=bqHRir7_BNlzwUqsEJf_bzfpsfTSlOpkOfAz65JRXCg,3797
@@ -885,9 +896,9 @@ lightning_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
 lightning_sdk/utils/dynamic.py,sha256=glUTO1JC9APtQ6Gr9SO02a3zr56-sPAXM5C3NrTpgyQ,1959
 lightning_sdk/utils/enum.py,sha256=h2JRzqoBcSlUdanFHmkj_j5DleBHAu1esQYUsdNI-hU,4106
 lightning_sdk/utils/resolve.py,sha256=idhRk4zg0jBOlySwKwjhsGgRorpsNVHOKOADOxMUYY0,6062
-lightning_sdk-0.1.55.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
-lightning_sdk-0.1.55.dist-info/METADATA,sha256=vUyBCQ2qsL1MRkz7vWH4AfWkWosAgiT7w-c2lHewz-0,4013
-lightning_sdk-0.1.55.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-lightning_sdk-0.1.55.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
-lightning_sdk-0.1.55.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
-lightning_sdk-0.1.55.dist-info/RECORD,,
+lightning_sdk-0.1.57.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-0.1.57.dist-info/METADATA,sha256=zK1E4W0HFXVjevGaq4MI9eqcvAYTOcToZAOFatRGOlE,4054
+lightning_sdk-0.1.57.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+lightning_sdk-0.1.57.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
+lightning_sdk-0.1.57.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-0.1.57.dist-info/RECORD,,