lightning-sdk 0.1.46__py3-none-any.whl → 0.1.47__py3-none-any.whl

lightning_sdk/mmt/base.py

@@ -1,7 +1,8 @@
 from abc import abstractmethod
-from typing import TYPE_CHECKING, Dict, Optional, Protocol, Tuple, Union
+from typing import TYPE_CHECKING, Dict, List, Optional, Protocol, Tuple, Union
 
 if TYPE_CHECKING:
+    from lightning_sdk.job.base import MachineDict
     from lightning_sdk.machine import Machine
     from lightning_sdk.organization import Organization
     from lightning_sdk.status import Status
@@ -41,6 +42,10 @@ class MMTMachine(Protocol):
         """The logs of the given machine."""
         ...
 
+    def dict(self) -> "MachineDict":
+        """Dict representation of the given machine."""
+        ...
+
 
 class _BaseMMT(_BaseJob):
     """Base interface to all job types."""
@@ -64,6 +69,7 @@ class _BaseMMT(_BaseJob):
         cloud_account_auth: bool = False,
         artifacts_local: Optional[str] = None,
         artifacts_remote: Optional[str] = None,
+        entrypoint: str = "sh -c",
         cluster: Optional[str] = None,  # deprecated in favor of cloud_account
     ) -> "_BaseMMT":
         """Run async workloads using a docker image across multiple machines.
@@ -98,6 +104,10 @@ class _BaseMMT(_BaseJob):
                 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.
         """
         from lightning_sdk.studio import Studio
 
@@ -148,6 +158,9 @@ class _BaseMMT(_BaseJob):
                     "Other jobs will automatically persist artifacts to the teamspace distributed filesystem."
                 )
 
+            if entrypoint != "sh -c":
+                raise ValueError("Specifying the entrypoint has no effect for jobs with Studio envs.")
+
         else:
             if studio is not None:
                 raise RuntimeError(
@@ -178,6 +191,7 @@ class _BaseMMT(_BaseJob):
             cloud_account_auth=cloud_account_auth,
             artifacts_local=artifacts_local,
             artifacts_remote=artifacts_remote,
+            entrypoint=entrypoint,
         )
         return inst
 
@@ -196,6 +210,7 @@ class _BaseMMT(_BaseJob):
         cloud_account_auth: bool = False,
         artifacts_local: Optional[str] = None,
         artifacts_remote: Optional[str] = None,
+        entrypoint: str = "sh -c",
     ) -> None:
         """Submit a new multi-machine job to the Lightning AI platform.
 
@@ -225,6 +240,9 @@ class _BaseMMT(_BaseJob):
                 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.
+                To use the pre-defined entrypoint of the provided image, set this to an empty string.
+                Only applicable when submitting docker jobs.
         """
 
     @property
@@ -283,6 +301,26 @@ class _BaseMMT(_BaseJob):
         """Logs of the rank 0 machine."""
         return self.machines[0].logs
 
+    def dict(
+        self
+    ) -> Dict[str, Union[str, "Studio", "Status", "Machine", None, List[Dict[str, Union[str, "Status", "Machine"]]]]]:
+        """Dict representation of this job."""
+        studio = self.studio
+
+        return {
+            "name": self.name,
+            "teamspace": f"{self.teamspace.owner.name}/{self.teamspace.name}",
+            "studio": studio.name if studio else None,
+            "image": self.image,
+            "command": self.command,
+            "status": self.status,
+            "machine": self.machine,
+            "machines": [
+                {"name": d["name"], "status": d["status"], "machine": d["machine"]}
+                for d in (x.dict() for x in self.machines)
+            ],
+        }
+
     @abstractmethod
     def _update_internal_job(self) -> None:
         pass

lightning_sdk/mmt/mmt.py

@@ -87,6 +87,7 @@ class MMT(_BaseMMT):
         cloud_account_auth: bool = False,
         artifacts_local: Optional[str] = None,
         artifacts_remote: Optional[str] = None,
+        entrypoint: str = "sh -c",
         cluster: Optional[str] = None,  # deprecated in favor of cloud_account
     ) -> "MMT":
         """Run async workloads using a docker image across multiple machines.
@@ -121,6 +122,9 @@ class MMT(_BaseMMT):
                 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.
+                To use the pre-defined entrypoint of the provided image, set this to an empty string.
+                Only applicable when submitting docker jobs.
         """
         ret_val = super().run(
             name=name,
@@ -166,6 +170,7 @@ class MMT(_BaseMMT):
         cloud_account_auth: bool = False,
         artifacts_local: Optional[str] = None,
         artifacts_remote: Optional[str] = None,
+        entrypoint: str = "sh -c",
     ) -> "MMT":
         """Submit a new multi-machine job to the Lightning AI platform.
 
@@ -195,6 +200,10 @@ class MMT(_BaseMMT):
                 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.
         """
         self._job = self._internal_mmt._submit(
             num_machines=num_machines,
@@ -270,6 +279,21 @@ class MMT(_BaseMMT):
     def link(self) -> str:
         return self._internal_mmt.link
 
+    @property
+    def studio(self) -> Optional["Studio"]:
+        """The studio used to submit the MMT."""
+        return self._internal_mmt.studio
+
+    @property
+    def image(self) -> Optional[str]:
+        """The image used to submit the MMT."""
+        return self._internal_mmt.image
+
+    @property
+    def command(self) -> str:
+        """The command the MMT is running."""
+        return self._internal_mmt.command
+
     def __getattr__(self, key: str) -> Any:
         """Forward the attribute lookup to the internal job implementation."""
         try:

lightning_sdk/mmt/v1.py

@@ -53,6 +53,7 @@ class _MMTV1(_BaseMMT):
         cloud_account_auth: bool = False,
         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.
 
@@ -82,10 +83,14 @@ class _MMTV1(_BaseMMT):
                 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.
         """
         if studio is None:
             raise ValueError("Studio is required for submitting jobs")
-        if image is not None or image_credentials is not None or cloud_account_auth:
+        if image is not None or image_credentials is not None or cloud_account_auth or entrypoint != "sh -c":
             raise ValueError("Image is not supported for submitting jobs")
 
         if artifacts_local is not None or artifacts_remote is not None:
@@ -176,6 +181,25 @@ class _MMTV1(_BaseMMT):
     def link(self) -> str:
         return f"https://lightning.ai/{self.teamspace.owner.name}/{self.teamspace.name}/studios/{self._job_api.get_studio_name(self._guaranteed_job)}/app?app_id=mmt&app_tab=Runs&job_name={self.name}"
 
+    @property
+    def image(self) -> Optional[str]:
+        """The image used to submit the job."""
+        # mmtv1 don't support images, so return None here
+        return None
+
+    @property
+    def studio(self) -> Optional["Studio"]:
+        """The studio used to submit the job."""
+        from lightning_sdk.studio import Studio
+
+        studio_name = self._job_api.get_studio_name(self._guaranteed_job)
+        return Studio(studio_name, teamspace=self.teamspace)
+
+    @property
+    def command(self) -> str:
+        """The command the job is running."""
+        return self._job_api.get_command(self._guaranteed_job)
+
     # the following and functions are solely to make the Work class function
     @property
     def _id(self) -> str:

lightning_sdk/mmt/v2.py

@@ -52,6 +52,7 @@ class _MMTV2(_BaseMMT):
         cloud_account_auth: bool = False,
         artifacts_local: Optional[str] = None,
         artifacts_remote: Optional[str] = None,
+        entrypoint: str = "sh -c",
     ) -> "_MMTV2":
         """Submit a new multi-machine job to the Lightning AI platform.
 
@@ -81,6 +82,10 @@ class _MMTV2(_BaseMMT):
                 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.
         """
         # Command is required if Studio is provided to know what to run
         # Image is mutually exclusive with Studio
@@ -113,6 +118,7 @@ class _MMTV2(_BaseMMT):
             cloud_account_auth=cloud_account_auth,
             artifacts_local=artifacts_local,
             artifacts_remote=artifacts_remote,
+            entrypoint=entrypoint,
         )
         self._job = submitted
         self._name = submitted.name
@@ -191,3 +197,25 @@ class _MMTV2(_BaseMMT):
     def link(self) -> str:
         # TODO: Since we don't have a UI for this yet, we can't have a link
         raise NotImplementedError
+
+    @property
+    def image(self) -> Optional[str]:
+        """The image used to submit the job."""
+        return self._job_api.get_image_name(self._guaranteed_job)
+
+    @property
+    def studio(self) -> Optional["Studio"]:
+        """The studio used to submit the job."""
+        from lightning_sdk.studio import Studio
+
+        studio_name = self._job_api.get_studio_name(self._guaranteed_job)
+
+        # if job was submitted with image, studio will be None
+        if not studio_name:
+            return None
+        return Studio(studio_name, teamspace=self.teamspace)
+
+    @property
+    def command(self) -> str:
+        """The command the job is running."""
+        return self._job_api.get_command(self._guaranteed_job)

lightning_sdk/status.py

@@ -4,10 +4,14 @@ from enum import Enum
 class Status(Enum):
     """Enum holding all possible studio status types."""
 
-    NotCreated = 1
-    Pending = 2
-    Running = 3
-    Stopping = 4
-    Stopped = 5
-    Completed = 6
-    Failed = 7
+    NotCreated = "NotCreated"
+    Pending = "Pending"
+    Running = "Running"
+    Stopping = "Stopping"
+    Stopped = "Stopped"
+    Completed = "Completed"
+    Failed = "Failed"
+
+    def __str__(self) -> str:
+        """String representation of the enum."""
+        return self.value

lightning_sdk/teamspace.py

@@ -1,6 +1,6 @@
 import warnings
 from pathlib import Path
-from typing import TYPE_CHECKING, List, Optional, Union
+from typing import TYPE_CHECKING, List, Optional, Tuple, Union
 
 from lightning_sdk.agents import Agent
 from lightning_sdk.api import TeamspaceApi
@@ -17,6 +17,8 @@ from lightning_sdk.utils.resolve import (
 )
 
 if TYPE_CHECKING:
+    from lightning_sdk.job import Job
+    from lightning_sdk.mmt import MMT
     from lightning_sdk.studio import Studio
 
 
@@ -126,6 +128,54 @@ class Teamspace:
         )
         return self.cloud_accounts
 
+    @property
+    def jobs(self) -> Tuple["Job", ...]:
+        from lightning_sdk.job import Job
+        from lightning_sdk.plugin import forced_v1
+
+        jobsv1, jobsv2 = self._teamspace_api.list_jobs(teamspace_id=self.id)
+
+        jobs = []
+
+        for j1 in jobsv1:
+            with forced_v1(Job):
+                # _fetch_job = False to prevent refetching on init since we already got it
+                job = Job(name=j1.name, teamspace=self, _fetch_job=False)
+            job._internal_job._job = j1
+            jobs.append(job)
+
+        for j2 in jobsv2:
+            # _fetch_job = False to prevent refetching on init since we already got it
+            job = Job(name=j2.name, teamspace=self, _fetch_job=False)
+            job._internal_job._job = j2
+            jobs.append(job)
+
+        return tuple(jobs)
+
+    @property
+    def multi_machine_jobs(self) -> Tuple["MMT", ...]:
+        from lightning_sdk.mmt import MMT
+        from lightning_sdk.plugin import forced_v1
+
+        mmtsv1, mmtsv2 = self._teamspace_api.list_mmts(teamspace_id=self.id)
+
+        mmts = []
+
+        for m1 in mmtsv1:
+            with forced_v1(MMT):
+                # _fetch_job = False to prevent refetching on init since we already got it
+                mmt = MMT(name=m1.name, teamspace=self, _fetch_job=False)
+            mmt._internal_mmt._job = m1
+            mmts.append(mmt)
+
+        for m2 in mmtsv2:
+            # _fetch_job = False to prevent refetching on init since we already got it
+            mmt = MMT(name=m2.name, teamspace=self, _fetch_job=False)
+            mmt._internal_mmt._job = m2
+            mmts.append(mmt)
+
+        return tuple(mmts)
+
     def __eq__(self, other: "Teamspace") -> bool:
         """Checks whether the provided other object is equal to this one."""
         return (

{lightning_sdk-0.1.46.dist-info → lightning_sdk-0.1.47.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning_sdk
-Version: 0.1.46
+Version: 0.1.47
 Summary: SDK to develop using Lightning AI Studios
 Author-email: Lightning-AI <justus@lightning.ai>
 License: MIT License

{lightning_sdk-0.1.46.dist-info → lightning_sdk-0.1.47.dist-info}/RECORD

@@ -1,47 +1,51 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
-lightning_sdk/__init__.py,sha256=uuMtBr9ocVg0IrTS9sa1JkMHSaKFKhoTWq7z3oF3m08,925
+lightning_sdk/__init__.py,sha256=8lax4DfCs2aWQNTxZwlBWy3Rp6jI3-IY89WASJupwZw,925
 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/machine.py,sha256=VdFXStR6ilYBEYuxgGWzcAw2TtW-nEQVsh6hz-2aaEw,750
+lightning_sdk/lit_registry.py,sha256=0zRFH-l61gHOslR6m0Rtetm-sWoUtN7240cvbxAx03I,1332
+lightning_sdk/machine.py,sha256=qutfVwQJHC02c2ygB_O5wrFI-Eq9zbfA0E1OPEYcCO0,861
 lightning_sdk/models.py,sha256=d27VAYUcbWKd4kuL_CqwCi3IguyjmKUR9EVWfXWTwmc,5606
 lightning_sdk/organization.py,sha256=WCfzdgjtvY1_A07DnxOpp74V2JR2gQwtXbIEcFDnoVU,1232
 lightning_sdk/owner.py,sha256=t5svD2it4C9pbSpVuG9WJL46CYi37JXNziwnXxhiU5U,1361
 lightning_sdk/plugin.py,sha256=nWFL1l6Q9DE8Lr2krD5qRhZljwmYm00R2eR1tYRb20s,14902
-lightning_sdk/status.py,sha256=kLDhN4-zdsGuZM577JMl1BbUIoF61bUOadW89ZAATFA,219
+lightning_sdk/status.py,sha256=lLGAuSvXBoXQFEEsEYwdCi0RcSNatUn5OPjJVjDtoM0,386
 lightning_sdk/studio.py,sha256=lezGs111RUFejLWgp4Urov5l6uiUmYJjtRK8D8EYFU8,17198
-lightning_sdk/teamspace.py,sha256=dKT-WrYF2xGP1C1bjOY2aYlEpkrvYkz2fXvtYVgogwo,11508
+lightning_sdk/teamspace.py,sha256=j6EghDJvxrYoWT1xMOA4qGAw3ezIpEytqGr-QlFnRXE,13247
 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=CYQLFLA89m3xQ-6Ss3UX4TDK6ZWRwmPGA5DjyJqW3RM,5578
 lightning_sdk/api/deployment_api.py,sha256=T480Nej7LqmtkAx8SBkPGQ5JxeyQ-GVIDqUCc7Z1yfk,21448
-lightning_sdk/api/job_api.py,sha256=KgXl0uO32Ja0AvxOASh-LihUPGERq2Fwy1rovXnq5Sg,11074
-lightning_sdk/api/mmt_api.py,sha256=texQJqSjbQNpfLLrumpvZ0MauPjmBlJAc8oSk3i46wk,6569
+lightning_sdk/api/job_api.py,sha256=L7O7zt3hy-s1UYDzO2FbLS7wtGXU3x8ZBDrtklQFZFc,11648
+lightning_sdk/api/lit_registry_api.py,sha256=peAZsJKVLvon7bVHq6aTYdcTVERiGoAZRQ5DzNean3Y,385
+lightning_sdk/api/mmt_api.py,sha256=rgMVDdlC3h-KqGI2FGp7Im99gnrHg5xx4FLwQqYWqxI,7136
 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=NYhD-Br2NIIn-1Noc8Q94TzWxEwFM1qCkf9RZhOqQu0,10321
+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=GbfTdqW1yV-fjyEwVebsbAdKQQjRmwT_fXUamrcnwAY,22534
 lightning_sdk/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lightning_sdk/cli/ai_hub.py,sha256=8oy6TogDiWnHuLT3cv33XEW7vPqXPA0dDMds8kX3Z4g,1649
 lightning_sdk/cli/download.py,sha256=nyQN3q1vZ0fg4_cfit8cKaokQ9VUd46l_TNcAQWkLwU,5996
-lightning_sdk/cli/entrypoint.py,sha256=Hl2Lm7-OS0kx_pyJyGe7Nii0Soc6HYe4r4xXKeJuC_o,1507
+lightning_sdk/cli/entrypoint.py,sha256=dR27MRdqTGnC_rQcP9RHV4Zq4GL0DpjHUlkD418-HZk,1576
 lightning_sdk/cli/exceptions.py,sha256=QUF3OMAMZwBikvlusimSHSBjb6ywvHpfAumJBEaodSw,169
 lightning_sdk/cli/legacy.py,sha256=ocTVNwlsLRS5aMjbMkwFPjT3uEYvS8C40CJ0PeRRv8g,4707
+lightning_sdk/cli/list.py,sha256=7xKZI3xnJmZo_7cWWdC7_n3XHq2sOb0EwgXNvFkp3vI,2106
 lightning_sdk/cli/run.py,sha256=B6ttd9SKg373ngug-lj74CEcuEoxwz-P6nUBVnQeijI,10836
 lightning_sdk/cli/serve.py,sha256=UaXhGHU6nbAzrnVigSKOTrMjLwSs-sjyhuJCdVUBwzc,8722
 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=H9OyipYTYAQ9Mzy2e8jtoaa-B34-uXHbTQTzY2Vmhv4,9078
 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=WHd2jz1qEeQWg2ljphlJMwoEVU6qtSBjnO9hMScLF7E,13383
-lightning_sdk/job/job.py,sha256=Z6W3qeA2_aG9X9qCju7hM-OBdE5aBsWw4daLWLsfLJk,11585
-lightning_sdk/job/v1.py,sha256=Ff1iTvZqa1J90eIex6_xDjwAxJztHtUSTmjihFwq5Vg,9060
-lightning_sdk/job/v2.py,sha256=mLEgSbog3QiLKsk1-9pWNWA_yYuQzvNfkWt8Mt2Y598,8720
-lightning_sdk/job/work.py,sha256=_L1eF9L0dW_BI17Wo0HwgoOkwwd48vFkqBD0uOt_rJk,2043
+lightning_sdk/job/base.py,sha256=bEhnFHrl3Ct5RNVqUCJRzzoF8NgXR6CMLyQiXBMIcSE,15526
+lightning_sdk/job/job.py,sha256=ZEyzlYtVCFeQJSiv0u6T5ShH5DVQZ3AGYnZ8dKoVAUo,12666
+lightning_sdk/job/v1.py,sha256=pmu-adVC0fh4jKwXvjBQb9iIDKPuC5eOo0a2BiBq1fw,10184
+lightning_sdk/job/v2.py,sha256=SakYCmTMnG_uP35Fb-zwWoiO0LHNqnCKd-pSvKsg564,9860
+lightning_sdk/job/work.py,sha256=pe1kWU3_GPuN5QlKIDOx_DKyzP1ARhJqY-ACKmklo8U,2313
 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
@@ -49,7 +53,7 @@ lightning_sdk/lightning_cloud/login.py,sha256=NSC53eJ24g59dySI9zRDl_-antxiflKDkn
 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=d_wz1rNzj1beoZ5OODYNCSv3gK12dD70bRb3DHXq3A8,85840
+lightning_sdk/lightning_cloud/openapi/__init__.py,sha256=vM_5AhUer_Oylf_q1Z6JrWGMsi8Gq_YiJ_lqHwZ9B5g,85952
 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,7 +77,7 @@ lightning_sdk/lightning_cloud/openapi/api/lightningapp_v2_service_api.py,sha256=
 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=TSFWpOJkyh5s4RB1VLwIjtJjygxgvXetxEJiuspzv4c,9914
+lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py,sha256=OzlfVFyDe0-K87mTtaCGQZbdtePKVvgNfvQUqTViOCk,14562
 lightning_sdk/lightning_cloud/openapi/api/models_store_api.py,sha256=JlBPgAPnP_4uzN3orcp21VdRxMYkL7m4IbSnLqGZEsw,93092
 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
@@ -86,7 +90,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=kR2Do68Mygtce5kvK7mOI_MgOXDzZ6r783d701aqLvE,80882
+lightning_sdk/lightning_cloud/openapi/models/__init__.py,sha256=llwvhKbCskiT1VVXVxCfOIfJ0cw15qHFh3TLtr_TWw0,80994
 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
@@ -356,6 +360,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_delete_cluster_capacity_reservat
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_cluster_encryption_key_response.py,sha256=F6QSFYZM8j9rDtOpwzbVEH17NTCx5qSlvT_UjcdeIAE,3106
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_cluster_proxy_response.py,sha256=0YtYHebFVutcWQMuY9To1Eh7zfRe9wFxb9RkLL8Nav0,3058
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_cluster_response.py,sha256=CCQd_OpnjAPaTP0Q3Uxw8nba58ZTnmkF-o2CEfwa9ec,3028
+lightning_sdk/lightning_cloud/openapi/models/v1_delete_container_response.py,sha256=6lYTA3fbtx1cTFjAtagGYa_42t1-wAwspT2djmZCbhY,3040
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_conversation_response.py,sha256=9X3MUKE0C8sqYP48lm-f0aN0AGe3I4H7IDUSbHprfa8,3058
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_data_connection_response.py,sha256=DsGzX9XH4d9Zaz8MPqrfN90oYzYjxK3swK4Styu7SbE,3070
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_dataset_response.py,sha256=3e9VCGV7FjV8qzvsXU3-SZi7Lj72Bc-qJ07x-69Ek7U,3028
@@ -397,14 +402,14 @@ lightning_sdk/lightning_cloud/openapi/models/v1_delete_user_slurm_job_response.p
 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=F4VmxQKXAYrRDlHMbDr7Ucj5eUCJ5QC2BXtO2orxIDA,6495
+lightning_sdk/lightning_cloud/openapi/models/v1_deployment_api.py,sha256=3nQ28vu8i_TWYEoyuvbxFCd-o2mWPR3SDqs2qPJw6aQ,7108
 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
 lightning_sdk/lightning_cloud/openapi/models/v1_deployment_performance.py,sha256=N-v8pFBVvJIAt0opbygd33R014Xeef3hoO4-ksT2yn4,10336
 lightning_sdk/lightning_cloud/openapi/models/v1_deployment_release.py,sha256=7sFF4w_XHV-bYPUGNRaADx2JInR8gxzPCmJ45NBRVfU,9430
 lightning_sdk/lightning_cloud/openapi/models/v1_deployment_spec.py,sha256=v1p0ryc6KgwRwXSJYjFLe4jzczpOMSiMws3JLjEoxGY,5922
-lightning_sdk/lightning_cloud/openapi/models/v1_deployment_state.py,sha256=sJAFZffnaz-NPH9hy86z3XH3_EY0-bZzh0ojWPMSAfY,3203
+lightning_sdk/lightning_cloud/openapi/models/v1_deployment_state.py,sha256=jDfHlwqCyzX2GK_QmlkiBHf5YaQMST7fJqEIRI3tF9c,3248
 lightning_sdk/lightning_cloud/openapi/models/v1_deployment_status.py,sha256=7D9Ux2N5LWmNXxrynoXPThExLD8FAVOzIVtIKLqYQfo,8939
 lightning_sdk/lightning_cloud/openapi/models/v1_deployment_strategy.py,sha256=aewPM7IDeAEEx2HoMySQ2hcIrme2X6UvEENih_kzjl0,4593
 lightning_sdk/lightning_cloud/openapi/models/v1_deployment_template.py,sha256=WuJWe8Wg5W6YqZUCpIEBoUJ2uwtMrTMizAd-VED2EV4,21585
@@ -490,7 +495,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_get_user_notification_preference
 lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py,sha256=z-3_OOHg830ETJPAnIS69yHF1LJcIRgjzVs2qForT2A,30480
 lightning_sdk/lightning_cloud/openapi/models/v1_get_user_storage_breakdown_response.py,sha256=igCbjaeZXEbiPAa9cUjwG4cPUivt-7MiiWYikELHwsw,9983
 lightning_sdk/lightning_cloud/openapi/models/v1_get_user_storage_response.py,sha256=0k1scTY1IAeKr3p0r7eV_xWRE7Nx54doTGygiQQPkbM,6879
-lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py,sha256=HVUNjM1zdc1ukUgAFHkmfQhmjwireAQ_3MzFeoFuj1w,13054
+lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py,sha256=gYnzf2hVpl7Kc5Fe8EBlSx2fJRBwDnsvkg4bCMkFfek,11861
 lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1_status.py,sha256=Md6rMiUCMY4oDVE2IYaALD_I0o2PQIKF-9628IGmTyY,3791
 lightning_sdk/lightning_cloud/openapi/models/v1_gpu_system_metrics.py,sha256=io_MkTCTYi6qmYw8KqFZf2zHx8jgBsPKsEwqjr2BG0U,7722
 lightning_sdk/lightning_cloud/openapi/models/v1_header.py,sha256=6GR8AL-UcMa3mH-PLtd7WWDZiG0Zi3AYKFw19AAasZc,4831
@@ -505,7 +510,7 @@ 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=N8fw267-dYxPDj7JGQicHZ86cIcCotK4ZW9tuWZ81Fw,19070
+lightning_sdk/lightning_cloud/openapi/models/v1_job.py,sha256=A0JgsixIYkVocVvuXsSZ4rdcV0npU-Llor_XWSE8Ink,19960
 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
@@ -715,7 +720,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_request_cluster_access_request.p
 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
 lightning_sdk/lightning_cloud/openapi/models/v1_resource_tag.py,sha256=bC1Arm6rDIbnNzcAa9eKt2CYPG_LXrM2mlY1Kswc2Po,4827
-lightning_sdk/lightning_cloud/openapi/models/v1_resource_visibility.py,sha256=-STB2GFK1ZZ82HLqWZRTqxaRSEBul23adF5qvXmlYfU,3746
+lightning_sdk/lightning_cloud/openapi/models/v1_resource_visibility.py,sha256=6ig5Yw6MdcUFV6m7M5JOw37Zn5TR4ENlgFgmqsgLvuk,4494
 lightning_sdk/lightning_cloud/openapi/models/v1_resources.py,sha256=GA2kSAecFOIQwwpfslPdHXlj06E8WTrHL3loOcKDUXQ,7476
 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
@@ -804,7 +809,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=SFkC8UWnQ0g5t-piIjBdfSW4FeOFOmO5DIIshug_G8g,74799
+lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py,sha256=Dch5r7uQWI98sBKronVzlelWQnBeU2utrh8PoYIn6k0,75514
 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
@@ -833,15 +838,15 @@ lightning_sdk/lightning_cloud/source_code/logs_socket_api.py,sha256=ngpv70IayG8M
 lightning_sdk/lightning_cloud/source_code/tar.py,sha256=mSKRWEX1SIAbXFcESNrQAqWSVULv4G0jgcvR_o2KJ5g,5813
 lightning_sdk/lightning_cloud/source_code/uploader.py,sha256=CQrWV6JNDiglyRfhc42ht4GWFtgRz_fBxw4tOCERzco,3207
 lightning_sdk/lightning_cloud/utils/__init__.py,sha256=kSNKoybELDJHs9WeCf279V5JtzOf1VqfKODlKYr9uRo,115
-lightning_sdk/lightning_cloud/utils/data_connection.py,sha256=tTCe8ootBf020w5mgqaYFijSI0FwTYJqkbG_m0P6ZBo,6775
+lightning_sdk/lightning_cloud/utils/data_connection.py,sha256=YpyQlJFGtgZsuCdjv_0iuyKh-6NzU-6rlzSEBBXALAs,6751
 lightning_sdk/lightning_cloud/utils/dataset.py,sha256=4nUspe8iAaRPgSYpXA2uAQCgydm78kJzhOIx3C9qKls,2011
 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=V_ysuSjddXksCjkpGy-YVqD9j8b0jA6hkfFz0oW-Jn0,12768
-lightning_sdk/mmt/mmt.py,sha256=Br73TO7TrEdx7VH80CA_gCGDgz3JX4p16R9DdrwQshk,12307
-lightning_sdk/mmt/v1.py,sha256=Ovzwc-mm3PCsyVb8XfvbWUSFGZEI8E5NKayQB1pXvwg,8159
-lightning_sdk/mmt/v2.py,sha256=mUHnBN33UOfD7LX5akGShG7W1q5vGASkVaOTPhHhpDI,8353
+lightning_sdk/mmt/base.py,sha256=yXmussF0qEh7S9KeYcKo7xzBlIAg1DCjgdf1iWc-L6M,14489
+lightning_sdk/mmt/mmt.py,sha256=fJL_cxXnoN10NZXMqwbAOlN7f_TULdZjmz7PGoM7GXo,13369
+lightning_sdk/mmt/v1.py,sha256=1hMJVXMcmG1k4zM8sN44uOddHkuJnX1Wjl_WVz-Ox5Q,9166
+lightning_sdk/mmt/v2.py,sha256=RY5W172aqVqcKbXJCdY5mdk7r201IAsxqtwG1KuEhP4,9474
 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
@@ -850,9 +855,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=RWvlOWLHjaHhR0W0zT3mN719cbzhFfYCKBss38zfv3k,5783
-lightning_sdk-0.1.46.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
-lightning_sdk-0.1.46.dist-info/METADATA,sha256=i5xIAdGz0dkYTDegs6kjV1vsZwMC8NQTtYs2il_jK0U,4031
-lightning_sdk-0.1.46.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-lightning_sdk-0.1.46.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
-lightning_sdk-0.1.46.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
-lightning_sdk-0.1.46.dist-info/RECORD,,
+lightning_sdk-0.1.47.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-0.1.47.dist-info/METADATA,sha256=mgPc8AxO4hg3mbNN21lwLtjkrF9JtSjzgUNU2oTKRns,4031
+lightning_sdk-0.1.47.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+lightning_sdk-0.1.47.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
+lightning_sdk-0.1.47.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-0.1.47.dist-info/RECORD,,