lightning-sdk 0.1.49__py3-none-any.whl → 0.1.51__py3-none-any.whl

lightning_sdk/mmt/v1.py

@@ -43,7 +43,7 @@ class _MMTV1(_BaseMMT):
     def _submit(
         self,
         num_machines: int,
-        machine: "Machine",
+        machine: Union["Machine", str],
         command: Optional[str] = None,
         studio: Optional["Studio"] = None,
         image: Optional[str] = None,
@@ -52,9 +52,10 @@ class _MMTV1(_BaseMMT):
         cloud_account: Optional[str] = None,
         image_credentials: Optional[str] = None,
         cloud_account_auth: bool = False,
+        entrypoint: str = "sh -c",
+        path_mappings: Optional[Dict[str, str]] = None,
         artifacts_local: Optional[str] = None,
         artifacts_remote: Optional[str] = None,
-        entrypoint: str = "sh -c",
     ) -> "_MMTV1":
         """Submit a new multi-machine job to the Lightning AI platform.
 
@@ -88,6 +89,8 @@ class _MMTV1(_BaseMMT):
                 just runs the provided command in a standard shell.
                 To use the pre-defined entrypoint of the provided image, set this to an empty string.
                 Only applicable when submitting docker jobs.
+            path_mappings: The mappings from data connection inside your container (not supported)
+
         """
         if studio is None:
             raise ValueError("Studio is required for submitting jobs")
@@ -166,7 +169,7 @@ class _MMTV1(_BaseMMT):
         return f"/teamspace/jobs/{self.name}/snapshot"
 
     @property
-    def machine(self) -> "Machine":
+    def machine(self) -> Union["Machine", str]:
         """Returns the machine type this job is running on."""
         return self.machines[0].machine
 

lightning_sdk/mmt/v2.py

@@ -42,7 +42,7 @@ class _MMTV2(_BaseMMT):
     def _submit(
         self,
         num_machines: int,
-        machine: "Machine",
+        machine: Union["Machine", str],
         command: Optional[str] = None,
         studio: Optional["Studio"] = None,
         image: Optional[str] = None,
@@ -51,9 +51,10 @@ class _MMTV2(_BaseMMT):
         cloud_account: Optional[str] = None,
         image_credentials: Optional[str] = None,
         cloud_account_auth: bool = False,
-        artifacts_local: Optional[str] = None,
-        artifacts_remote: Optional[str] = None,
         entrypoint: str = "sh -c",
+        path_mappings: Optional[Dict[str, str]] = None,
+        artifacts_local: Optional[str] = None,  # deprecated in favor of path_mappings
+        artifacts_remote: Optional[str] = None,  # deprecated in favor of path_mappings
     ) -> "_MMTV2":
         """Submit a new multi-machine job to the Lightning AI platform.
 
@@ -73,20 +74,19 @@ class _MMTV2(_BaseMMT):
                 This should be the name of the respective credentials secret created on the Lightning AI platform.
             cloud_account_auth: Whether to authenticate with the cloud account to pull the image.
                 Required if the registry is part of a cloud provider (e.g. ECR).
-            artifacts_local: The path of inside the docker container, you want to persist images from.
-                CAUTION: When setting this to "/", it will effectively erase your container.
-                Only supported for jobs with a docker image compute environment.
-            artifacts_remote: The remote storage to persist your artifacts to.
-                Should be of format <CONNECTION_TYPE>:<CONNECTION_NAME>:<PATH_WITHIN_CONNECTION>.
-                PATH_WITHIN_CONNECTION hereby is a path relative to the connection's root.
-                E.g. efs:data:some-path would result in an EFS connection named `data` and to the path `some-path`
-                within it.
-                Note that the connection needs to be added to the teamspace already in order for it to be found.
-                Only supported for jobs with a docker image compute environment.
             entrypoint: The entrypoint of your docker container. Defaults to `sh -c` which
                 just runs the provided command in a standard shell.
                 To use the pre-defined entrypoint of the provided image, set this to an empty string.
                 Only applicable when submitting docker jobs.
+            path_mappings: Dictionary of path mappings. The keys are the path inside the container whereas the value
+                represents the data-connection name and the path inside that connection.
+                Should be of form
+                    {
+                        "<CONTAINER_PATH_1>": "<CONNECTION_NAME_1>:<PATH_WITHIN_CONNECTION_1>",
+                        "<CONTAINER_PATH_2>": "<CONNECTION_NAME_2>"
+                    }
+                If the path inside the connection is omitted it's assumed to be the root path of that connection.
+                Only applicable when submitting docker jobs.
         """
         # Command is required if Studio is provided to know what to run
         # Image is mutually exclusive with Studio
@@ -117,9 +117,10 @@ class _MMTV2(_BaseMMT):
             env=env,
             image_credentials=image_credentials,
             cloud_account_auth=cloud_account_auth,
+            entrypoint=entrypoint,
+            path_mappings=path_mappings,
             artifacts_local=artifacts_local,
             artifacts_remote=artifacts_remote,
-            entrypoint=entrypoint,
         )
         self._job = submitted
         self._name = submitted.name
@@ -169,7 +170,7 @@ class _MMTV2(_BaseMMT):
         raise NotImplementedError
 
     @property
-    def machine(self) -> "Machine":
+    def machine(self) -> Union["Machine", str]:
         """Returns the machine type this job is running on."""
         return self._job_api._get_job_machine_from_spec(self._guaranteed_job.spec)
 

lightning_sdk/models.py

@@ -121,10 +121,11 @@ def download_model(
             progress_bar=progress_bar,
         )
     except ApiException as e:
-        # if we get an error, check if the teamspace actually exists (and print the list)
-        # TODO: ideally this would match a specific error about teamspace not being found
-        _ = _get_teamspace(name=teamspace_name, organization=teamspace_owner_name)
-        raise e
+        if e.status == 404:
+            raise RuntimeError(
+                f"Model '{name}' not found. Either the model doesn't exist or you don't have access to it."
+            ) from None
+        raise RuntimeError(f"Error downloading model. Status code: {e.status}.") from None
 
 
 def upload_model(

lightning_sdk/utils/resolve.py

@@ -187,3 +187,10 @@ def _parse_model_and_version(name: str) -> Tuple[str, str]:
         "Model version is expected to be in the format `entity/modelname:version` separated by a"
         f" single colon, but got: {name}"
     )
+
+
+def in_studio() -> bool:
+    """Returns true if inside a studio, else false."""
+    has_cloudspace_id = bool(os.getenv("LIGHTNING_CLOUD_SPACE_ID", None))
+    is_interactive = os.getenv("LIGHTNING_INTERACTIVE", "false") == "true"
+    return has_cloudspace_id and is_interactive

{lightning_sdk-0.1.49.dist-info → lightning_sdk-0.1.51.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning_sdk
-Version: 0.1.49
+Version: 0.1.51
 Summary: SDK to develop using Lightning AI Studios
 Author-email: Lightning-AI <justus@lightning.ai>
 License: MIT License
@@ -45,9 +45,9 @@ Requires-Dist: tqdm
 Requires-Dist: fire
 Requires-Dist: simple-term-menu
 Requires-Dist: lightning-utilities
+Requires-Dist: docker
 Provides-Extra: serve
 Requires-Dist: litserve>=0.2.5; extra == "serve"
-Requires-Dist: docker; extra == "serve"
 
 # Lightning SDK
 

{lightning_sdk-0.1.49.dist-info → lightning_sdk-0.1.51.dist-info}/RECORD

@@ -1,12 +1,12 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
-lightning_sdk/__init__.py,sha256=pghaQ6_YR2SmOR4N7VM5mlRGqecB5VgWJpqV0JMLxiE,970
+lightning_sdk/__init__.py,sha256=JRvKHTz5yFm-BcQqufNYC2Hz8AQAF1L5sdeIu-3Ad2c,970
 lightning_sdk/agents.py,sha256=ly6Ma1j0ZgGPFyvPvMN28JWiB9dATIstFa5XM8pMi6I,1577
 lightning_sdk/ai_hub.py,sha256=kBjtmrzVHPCgqtV_TrSNkuf4oT2DLm8SYRTz4iTQmmY,6624
 lightning_sdk/constants.py,sha256=ztl1PTUBULnqTf3DyKUSJaV_O20hNtUYT6XvAYIrmIk,749
-lightning_sdk/helpers.py,sha256=RnQwUquc_YPotjh6YXOoJvZs8krX_QFhd7kGv4U_spQ,1844
-lightning_sdk/lit_container.py,sha256=Yh24NmVBsixKSYIZ_p0QqcZ0kjJO58OvXYDLC7ptx8w,2954
+lightning_sdk/helpers.py,sha256=odfwLvceDbSuwj2nK21o3ER7D-h1j5cppHs1S4ThT18,1835
+lightning_sdk/lit_container.py,sha256=423XYYpzoGppv200kMbD9dB-iqP6xdHj7q5kejJLDKE,3778
 lightning_sdk/machine.py,sha256=qutfVwQJHC02c2ygB_O5wrFI-Eq9zbfA0E1OPEYcCO0,861
-lightning_sdk/models.py,sha256=d27VAYUcbWKd4kuL_CqwCi3IguyjmKUR9EVWfXWTwmc,5606
+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=nWFL1l6Q9DE8Lr2krD5qRhZljwmYm00R2eR1tYRb20s,14902
@@ -18,18 +18,18 @@ lightning_sdk/api/__init__.py,sha256=Qn2VVRvir_gO7w4yxGLkZY-R3T7kdiTPKgQ57BhIA9k
 lightning_sdk/api/agents_api.py,sha256=G47TbFo9kYqnBMqdw2RW-lfS1VAUBSXDmzs6fpIEMUs,4059
 lightning_sdk/api/ai_hub_api.py,sha256=CYQLFLA89m3xQ-6Ss3UX4TDK6ZWRwmPGA5DjyJqW3RM,5578
 lightning_sdk/api/deployment_api.py,sha256=T480Nej7LqmtkAx8SBkPGQ5JxeyQ-GVIDqUCc7Z1yfk,21448
-lightning_sdk/api/job_api.py,sha256=eu0YKA27m8nEX6d9VzyMvQsl_qnDwtDJwbD4uT1o2Ks,12230
-lightning_sdk/api/lit_container_api.py,sha256=wMX1FdTGiia-fqYFvJLH-Rf6eJIRJ9ggqrI84DtSAFI,1875
-lightning_sdk/api/mmt_api.py,sha256=NHLFaqHx0v2uwq9JEUkYPpDAf-L6EJkJS7nulj66ASA,7322
+lightning_sdk/api/job_api.py,sha256=3ohMeDjeOaQAWU2qwbg4v5cDWF3U915hECMODxTdN6M,12541
+lightning_sdk/api/lit_container_api.py,sha256=zgmLcX6883nCWPLNMEANYTLB4zXAbMwXvGacHTzN5wI,3371
+lightning_sdk/api/mmt_api.py,sha256=km27mnZ7CT_TYa3ofjVrPlidjRMJk32pnznHxjVf1OM,7633
 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/user_api.py,sha256=sL7RIjjtmZmvCZWx7BBZslhj1BeNh4Idn-RVcdmf7M0,2598
-lightning_sdk/api/utils.py,sha256=VG3sCxWjn6MHONYghA73PZH-ErgqsHyj1ZMU6qzBCFk,22762
+lightning_sdk/api/utils.py,sha256=6j5bzUgfwTzUyWHlDcjG8w2rlNtqf3L7VocTzL466tU,24432
 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=TePfbCUKR2Bzq3D-avAX6AATQAR0GfFIxj_Qw3qSczo,2431
-lightning_sdk/cli/download.py,sha256=nyQN3q1vZ0fg4_cfit8cKaokQ9VUd46l_TNcAQWkLwU,5996
+lightning_sdk/cli/download.py,sha256=Fw_Gqhkb4668V5GgmDmM2q36Oogbj7q3f4VWaaqbBeY,6886
 lightning_sdk/cli/entrypoint.py,sha256=dybKJ9lTs7ixzaVRLGk4Gv1b11j3VKlAJr5B7itfDRA,2120
 lightning_sdk/cli/exceptions.py,sha256=QUF3OMAMZwBikvlusimSHSBjb6ywvHpfAumJBEaodSw,169
 lightning_sdk/cli/inspect.py,sha256=VeiazH-9DNn0SrQmP-ZM5JDdy8veQy5yxVe_Dj-KQWI,1349
@@ -38,28 +38,28 @@ lightning_sdk/cli/jobs_menu.py,sha256=6VqJecnyybYjdozj7MXH1gJehskJlwP8SoewQEXDz8
 lightning_sdk/cli/legacy.py,sha256=ocTVNwlsLRS5aMjbMkwFPjT3uEYvS8C40CJ0PeRRv8g,4707
 lightning_sdk/cli/list.py,sha256=gLfIf9iXPPyotqIy8MkWGJXWmGF8BV3jtmcZ8cCd3Ng,3856
 lightning_sdk/cli/mmts_menu.py,sha256=OAwWywYeFSdmXBLR7raZlL7tpAhZld4r71LUHvmnZdA,2175
-lightning_sdk/cli/run.py,sha256=4hcPuawGPVoHMoLUw6uwTNv1qPd4Js-PomUJgjmLkfw,11806
-lightning_sdk/cli/serve.py,sha256=UaXhGHU6nbAzrnVigSKOTrMjLwSs-sjyhuJCdVUBwzc,8722
+lightning_sdk/cli/run.py,sha256=nfT_oq_3zvi6oPF-edPNUA82yOwGSOTCSx81eCnlPI8,14107
+lightning_sdk/cli/serve.py,sha256=TzDbaUtE5Bj-AFucnNKqk40TkIagUtoRPfq-7RFT16g,8553
 lightning_sdk/cli/stop.py,sha256=qBlVaYAJGxiNSwskxqB7zizTVPIpBGlJ4c7kIXII7iw,1373
 lightning_sdk/cli/studios_menu.py,sha256=0kQGqGel8gAbpdJtjOM1a6NEat_TnIqRNprNn8QiK58,3236
 lightning_sdk/cli/teamspace_menu.py,sha256=GVTrMAqxxL3INX83alo5wybnrl6rBBKENo01ZNkWPik,3753
-lightning_sdk/cli/upload.py,sha256=pn7yIq98e_gMoaFhETHfiXai_15qdyG6t1hsdjN-oWg,10589
+lightning_sdk/cli/upload.py,sha256=cvirl8lfPiGV4THa5akeRy1KDoPx7Klj7tTrvWePwLc,11501
 lightning_sdk/deployment/__init__.py,sha256=BLu7_cVLp97TYxe6qe-J1zKUSZXAVcvCjgcA7plV2k4,497
 lightning_sdk/deployment/deployment.py,sha256=Dp15pn8rFAfMfaDhKn0v3bphFuvLgkPFs3KSNxW6eyc,15472
 lightning_sdk/job/__init__.py,sha256=1MxjQ6rHkyUHCypSW9RuXuVMVH11WiqhIXcU2LCFMwE,64
-lightning_sdk/job/base.py,sha256=MT1p8RGIhJ9j7LVNTOfk6PTokP6GCdLe4U83vQ_P9dw,16272
-lightning_sdk/job/job.py,sha256=JSd2UUvBJof_ZVaO-FWWbv-Jml7othL8ZSuP6F9Sp7Y,13425
-lightning_sdk/job/v1.py,sha256=ncQcP7pBLUmyLSYwRNM6DVMhxVzRwIfs4EhmSJOe5aI,10202
-lightning_sdk/job/v2.py,sha256=gxrVjFOeHgvEBARE2w5nSUfNSuNEgdNl_RVGIvNPvlM,10180
-lightning_sdk/job/work.py,sha256=pe1kWU3_GPuN5QlKIDOx_DKyzP1ARhJqY-ACKmklo8U,2313
+lightning_sdk/job/base.py,sha256=dBVAiK8mku6Qa8CfbQ7d3ZqaTtW7bf_xsfJM2h246dU,16957
+lightning_sdk/job/job.py,sha256=TyueqMyci4rxmbDhqB3WUlkYsOHRFwl4lQrNv87I5As,13338
+lightning_sdk/job/v1.py,sha256=F9XqJTVx2JomxM6jYXqTpE394zTGZVGWv1JtAetj3uY,10378
+lightning_sdk/job/v2.py,sha256=7tamPEbgK6JuqsC3BFMi8Z2-wWR7qvnhjvdo0hTbNFw,10078
+lightning_sdk/job/work.py,sha256=Xm5byWTcOxYI8zZ3Sz6P8HxBTBAJ3S2KucndXA5KePQ,2332
 lightning_sdk/lightning_cloud/__init__.py,sha256=o91SMAlwr4Ke5ESe8fHjqXcj31_h7rT-MlFoXA-n2EI,173
 lightning_sdk/lightning_cloud/__version__.py,sha256=lOfmWHtjmiuSG28TbKQqd2B3nwmSGOlKVFwhaj_cRJk,23
 lightning_sdk/lightning_cloud/env.py,sha256=XZXpF4sD9jlB8DY0herTy_8XiUJuDVjxy5APjRD2_aU,1379
-lightning_sdk/lightning_cloud/login.py,sha256=NSC53eJ24g59dySI9zRDl_-antxiflKDknYKxUNcwzQ,7307
+lightning_sdk/lightning_cloud/login.py,sha256=02f3itTEMJ4wKpVxOfAAfWfigVeX8G51egCwaqcrEDQ,7637
 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=vLo-D5ix-ujPnwDyi69GY82YOTc5k3CLj4gqD2epNiw,86157
+lightning_sdk/lightning_cloud/openapi/__init__.py,sha256=XqMba2lpjS6djUj0l0vovAvT1MI0VcFgJkra6OvkIxs,86480
 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
@@ -77,13 +77,13 @@ lightning_sdk/lightning_cloud/openapi/api/deployment_templates_service_api.py,sh
 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=9ZS5ADZj9Oxz3Al6YyTZxCMBduQuQSYxCVyx9lDSKkU,29878
-lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py,sha256=dTIfytMp_azUb4-fzu2zrYXrCs3JLvjB4yf-28DquMk,163638
+lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py,sha256=cZktsDckrFVS9Qr-bl2xpU6JAMofo1owPhGdfMXdt8w,163850
 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_page_service_api.py,sha256=xf48UmWpzjs3woKVfFeyyqCLN13NA_qDepIwMw2SRUg,20457
-lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py,sha256=XJSRbTdtPDkq5XB0FDxK5Mbbf6KpSv7sRn1Bd4T1FVY,18996
+lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py,sha256=Glt-caamj5VZhM-v9RJhUBINZBbh52nv1eJ8awjKADE,24444
 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
@@ -96,7 +96,7 @@ 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=4SCpucxSLULv26jHjFJWoLXM7bySp2YeJnQayXZLWDU,47660
 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=wjdWKwf3zPIyaV7RpwB6QSqDHFX_sMh4ZD2DpfSujV8,66572
-lightning_sdk/lightning_cloud/openapi/models/__init__.py,sha256=Q6vEgJLooe1vC6Cp0TG5Olu58As0Wzz6l4dDCAFXgcs,81199
+lightning_sdk/lightning_cloud/openapi/models/__init__.py,sha256=txtixFYf4XP8qp2el7VMwPqYRgUAMTnnfimBK65vLfA,81522
 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
@@ -128,7 +128,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=PCVdZCbfcZwv-uTFEiJKUYb4Vgum72hDI97eSqbhv_s,16716
+lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py,sha256=mz0j73zJTOi2iGUrv0Q0Lv13L1lNuD9kY1GUCAsY4Qo,17363
 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
@@ -172,6 +172,7 @@ lightning_sdk/lightning_cloud/openapi/models/jobs_id_body2.py,sha256=ESSqHkNRBx9
 lightning_sdk/lightning_cloud/openapi/models/jobs_id_body3.py,sha256=iYsfZJt_IgARZnDRKnKrdqwHRLg5I285p1a-f_aDMcg,4644
 lightning_sdk/lightning_cloud/openapi/models/litloggermetrics_id_body.py,sha256=yZ9gb1EV-1UTLSuF_0T1FIyHuMcYEsoczc3x-qORFRA,5461
 lightning_sdk/lightning_cloud/openapi/models/litpages_id_body.py,sha256=-0ImUYoTU73XiLZBDGcqGmnSj3ZO7mDltXeXoDtBwfs,9811
+lightning_sdk/lightning_cloud/openapi/models/litregistry_lit_repo_name_body.py,sha256=KZY9WeqcTUgOyyh6fzNB5cD3BlcQT_NcPHdUsSRIzmc,4019
 lightning_sdk/lightning_cloud/openapi/models/loggermetrics_id_body.py,sha256=snEf_CsjDcd0kt0cFDPqDHCdAkoAjYCiisrBk4mAZR0,3763
 lightning_sdk/lightning_cloud/openapi/models/metrics_stream_id_loggerartifacts_body.py,sha256=MUMnHTkEmomIKq8f3fp8oP8jqQ1oE2i8K_TCxxtj3Q0,3781
 lightning_sdk/lightning_cloud/openapi/models/metricsstream_create_body.py,sha256=4xaorYmlhnHkKsnPIj38w7OqvG6oPzhu4DZwbNRWWws,11404
@@ -301,7 +302,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_cluster_names.py,sha256=Xs3mhVEi
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_proxy.py,sha256=4LbHvINs2CwQnuPg_uWw03GNNRT_wdqX42m_Zlqg2dY,5547
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_resource_tag.py,sha256=U6DfNNAIBZQ8SOF-2RXp2-CgmxaT1XJrNaCnZzaqjHg,4274
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_security_options.py,sha256=rr-Y_51425UxWZY9PEE9OW-m6mvqgmOP1yc3RB5FkIY,12571
-lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py,sha256=xGy5pWDcuKn1p90M4BGFlhAscswRMRmfRIRHdItR9aQ,17392
+lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py,sha256=92hMr_dTLE3eR0-nKsyeBFWF5-2UacNPjfqsUUEYzcs,18421
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_state.py,sha256=Mvl3dEhdtb7-rU4wUsR_-YOMpd7wE8uUAyXv3MiOcOk,3331
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_status.py,sha256=n9NhKdWg51q_7d4VM9Bsk5aakctU-o3R-0YmdzTsMW0,13267
 lightning_sdk/lightning_cloud/openapi/models/v1_cluster_tagging_options.py,sha256=E3uMncqfSA1IxGGAUsNYdfeXYkY4NNs6p7cLWBMoHMw,6283
@@ -407,8 +408,8 @@ 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=jlxDSFb2-frgy1zHIkmtrB0UoQvs3WvO6Pn5YlHq93Q,17607
-lightning_sdk/lightning_cloud/openapi/models/v1_deployment_api.py,sha256=3nQ28vu8i_TWYEoyuvbxFCd-o2mWPR3SDqs2qPJw6aQ,7108
+lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py,sha256=uGO8_1XLqnIkwmaBG5QLbc35Rl8vGwCEK0o4obWiOoI,18234
+lightning_sdk/lightning_cloud/openapi/models/v1_deployment_api.py,sha256=2s5n42kzaKEvSDyWPE2efEgvEFPAkptAGQohLfTs70M,7861
 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
 lightning_sdk/lightning_cloud/openapi/models/v1_deployment_metrics.py,sha256=WSs5FnpcLQvfsinsn4l5ZCmhX58CjMJUi8d6itB5DBk,4556
@@ -523,7 +524,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_job_health_check_config.py,sha25
 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=MwfESuUgK4fkPrsDqFHO67wFhOQc4P0pIKnJG0ehaNM,19716
+lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py,sha256=AwSo3KoarhsaLxP4ZOWCEhbihkr_-b8u_he-gGPPXSw,20580
 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
@@ -685,6 +686,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_organization.py,sha256=0sUmiM0aU
 lightning_sdk/lightning_cloud/openapi/models/v1_owner_type.py,sha256=DBEz-jE3e3PAt1kWVQ6LKjl3sWCkKEB4FFXyxSXkaZ4,3099
 lightning_sdk/lightning_cloud/openapi/models/v1_package_manager.py,sha256=31gbBiEQjOd_ToQ9gFklCAI_dtyxYl5dWYOS2al5iOs,3143
 lightning_sdk/lightning_cloud/openapi/models/v1_parameterization_spec.py,sha256=zQXkrKilutqMWRTTOi2_r15a9-I0XmVGTtiT0JD6Foc,6783
+lightning_sdk/lightning_cloud/openapi/models/v1_path_mapping.py,sha256=1ZI-QBCI6Zfa8FtGtFnR30v-6R-SgVIqI_IfQU_1bzk,5418
 lightning_sdk/lightning_cloud/openapi/models/v1_path_telemetry.py,sha256=8kErw3szKZW-IZ354XofR4UcJ4dZbzkmQ5vjWzGyCao,4082
 lightning_sdk/lightning_cloud/openapi/models/v1_phase_type.py,sha256=1RDAL-nvZBC9hsR4KNGa3gCuc1HtRu8G8cZjdOxyCN4,3071
 lightning_sdk/lightning_cloud/openapi/models/v1_plugin.py,sha256=mzcUs5AnfMrqMvIf7JO0DG8DkuCqYWd3pTmUXjedR_4,6883
@@ -792,6 +794,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_update_cluster_accelerators_resp
 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_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_project_cluster_accelerators_response.py,sha256=9NujcwHJzLFuj66q_AZVdyyZrFiy7_5pwOx89eFdS_U,3142
@@ -815,7 +818,7 @@ 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=BVJiD5oLzTBceVjdk2qS0_eA4xG4ANQzQAQNsoAYG-c,74282
+lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py,sha256=ZbebqSFWBe6KifRPUL6oMzoZsYmWZG6bOAaKxtlrzt8,74560
 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
@@ -851,10 +854,10 @@ lightning_sdk/lightning_cloud/utils/dataset.py,sha256=4nUspe8iAaRPgSYpXA2uAQCgyd
 lightning_sdk/lightning_cloud/utils/name_generator.py,sha256=MkciuA10332V0mcE2PxLIiwWomWE0Fm_gNGK01vwRr4,58046
 lightning_sdk/lightning_cloud/utils/network.py,sha256=axPgl8rhyPcPjxiztDxyksfxax3VNg2OXL5F5Uc81b4,406
 lightning_sdk/mmt/__init__.py,sha256=ExMu90-96bGBnyp5h0CErQszUGB1-PcjC4-R8_NYbeY,117
-lightning_sdk/mmt/base.py,sha256=KBqNMsbAQPGy5CPUCkZMbAGLnjyVG9H-o-xDAPQ-vng,14822
-lightning_sdk/mmt/mmt.py,sha256=v-dOMS1bqUGoGYIe45YZJc0aXPUDsUGNdRoeu8DMw60,14029
-lightning_sdk/mmt/v1.py,sha256=qdd8wrGlyg-75rYLGFY4-PfIg_O9fYvVRL-xryNg2U8,9347
-lightning_sdk/mmt/v2.py,sha256=97sIFx55MAgkaOq3pb5I8GB8pQoHg7HB8pezccH-vuw,9682
+lightning_sdk/mmt/base.py,sha256=B3HC-c82bPHprEZh1mhLCPCrCE8BOKqwIhY7xCF9CXg,15152
+lightning_sdk/mmt/mmt.py,sha256=vAzKIWhI9cXAkv5sBhOF0fKMvqdb4J1kZnF5FScejFM,13893
+lightning_sdk/mmt/v1.py,sha256=dROOXRWLzG2kUwGtvozB9Vywt_3QK8j4XFHVZFohWUM,9527
+lightning_sdk/mmt/v2.py,sha256=Em1XBoqViqUMKm-sshzdMcSH5UTtZZwbJcsgqY6-mw0,9625
 lightning_sdk/services/__init__.py,sha256=gSWUjccEhMI9CIWL_nbrFHUK2S6TM2725mEzrLMfK1Y,225
 lightning_sdk/services/file_endpoint.py,sha256=we5HC_o74J4Y6fSP_31jIizi_I_1FO_Rb2qblspD9eE,7855
 lightning_sdk/services/utilities.py,sha256=IeOx8hc3F8ZevHeKBysh08BXhJliTNzvKp1gwpEfdik,4087
@@ -862,10 +865,10 @@ lightning_sdk/services/finetune/__init__.py,sha256=5ewjo_bsk5hwvhGoLRdHBv1nP6Leo
 lightning_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 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=RWvlOWLHjaHhR0W0zT3mN719cbzhFfYCKBss38zfv3k,5783
-lightning_sdk-0.1.49.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
-lightning_sdk-0.1.49.dist-info/METADATA,sha256=tBtxBwoidv39GiVZ8kKllvqj9UEomokZZpD9_zlMSqs,4031
-lightning_sdk-0.1.49.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-lightning_sdk-0.1.49.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
-lightning_sdk-0.1.49.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
-lightning_sdk-0.1.49.dist-info/RECORD,,
+lightning_sdk/utils/resolve.py,sha256=idhRk4zg0jBOlySwKwjhsGgRorpsNVHOKOADOxMUYY0,6062
+lightning_sdk-0.1.51.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-0.1.51.dist-info/METADATA,sha256=RCGR1JfAwqKRGrnzblVnyrfmULZvKxL8d9eyDYYbs2A,4013
+lightning_sdk-0.1.51.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+lightning_sdk-0.1.51.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
+lightning_sdk-0.1.51.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-0.1.51.dist-info/RECORD,,