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

lightning_sdk/__init__.py

@@ -29,5 +29,5 @@ __all__ = [
     "AIHub",
 ]
 
-__version__ = "0.1.49"
+__version__ = "0.1.50"
 _check_version_and_prompt_upgrade(__version__)

lightning_sdk/api/job_api.py

@@ -1,11 +1,11 @@
 import time
-from typing import TYPE_CHECKING, Dict, List, Optional
+from typing import TYPE_CHECKING, Dict, List, Optional, Union
 from urllib.request import urlopen
 
 from lightning_sdk.api.utils import (
     _COMPUTE_NAME_TO_MACHINE,
-    _MACHINE_TO_COMPUTE_NAME,
     _create_app,
+    _machine_to_compute_name,
     remove_datetime_prefix,
 )
 from lightning_sdk.api.utils import (
@@ -120,7 +120,7 @@ class JobApiV1:
         studio_id: str,
         teamspace_id: str,
         cloud_account: str,
-        machine: Machine,
+        machine: Union[Machine, str],
         interruptible: bool,
     ) -> Externalv1LightningappInstance:
         """Creates an arbitrary app."""
@@ -130,7 +130,7 @@ class JobApiV1:
             teamspace_id=teamspace_id,
             cloud_account=cloud_account,
             plugin_type="job",
-            compute=_MACHINE_TO_COMPUTE_NAME[machine],
+            compute=_machine_to_compute_name(machine),
             name=name,
             entrypoint=command,
             interruptible=interruptible,
@@ -209,7 +209,7 @@ class JobApiV2:
         teamspace_id: str,
         studio_id: Optional[str],
         image: Optional[str],
-        machine: Machine,
+        machine: Union[Machine, str],
         interruptible: bool,
         env: Optional[Dict[str, str]],
         image_credentials: Optional[str],
@@ -223,7 +223,7 @@ class JobApiV2:
             for k, v in env.items():
                 env_vars.append(V1EnvVar(name=k, value=v))
 
-        instance_name = _MACHINE_TO_COMPUTE_NAME[machine]
+        instance_name = _machine_to_compute_name(machine)
 
         run_id = __GLOBAL_LIGHTNING_UNIQUE_IDS_STORE__[studio_id] if studio_id is not None else ""
 

lightning_sdk/api/lit_container_api.py

@@ -10,6 +10,14 @@ class LitContainerApi:
     def __init__(self) -> None:
         self._client = LightningClient(max_tries=3)
 
+        import docker
+
+        try:
+            self._docker_client = docker.from_env()
+            self._docker_client.ping()
+        except docker.errors.DockerException as e:
+            raise RuntimeError(f"Failed to connect to Docker daemon: {e!s}. Is Docker running?") from None
+
     def list_containers(self, project_id: str) -> List:
         project = self._client.lit_registry_service_get_lit_project_registry(project_id)
         return project.repositories
@@ -24,19 +32,24 @@ class LitContainerApi:
         import docker
 
         try:
-            client = docker.from_env()
-            client.ping()
-        except docker.errors.DockerException as e:
-            raise RuntimeError(f"Failed to connect to Docker daemon: {e!s}. Is Docker running?") from None
-
-        try:
-            client.images.get(container)
+            self._docker_client.images.get(container)
         except docker.errors.ImageNotFound:
             raise ValueError(f"Container {container} does not exist") from None
 
         registry_url = _get_registry_url()
         repository = f"{registry_url}/lit-container/{teamspace.owner.name}/{teamspace.name}/{container}"
-        tagged = client.api.tag(container, repository, tag)
+        tagged = self._docker_client.api.tag(container, repository, tag)
         if not tagged:
             raise ValueError(f"Could not tag container {container} with {repository}:{tag}")
-        return client.api.push(repository, stream=True, decode=True)
+        return self._docker_client.api.push(repository, stream=True, decode=True)
+
+    def download_container(self, container: str, teamspace: Teamspace, tag: str) -> Generator[str, None, None]:
+        import docker
+
+        registry_url = _get_registry_url()
+        repository = f"{registry_url}/lit-container/{teamspace.owner.name}/{teamspace.name}/{container}"
+        try:
+            self._docker_client.images.pull(repository, tag=tag)
+        except docker.errors.APIError as e:
+            raise ValueError(f"Could not pull container {container} from {repository}:{tag}") from e
+        return self._docker_client.api.tag(repository, container, tag)

lightning_sdk/api/mmt_api.py

@@ -1,12 +1,12 @@
 import json
 import time
-from typing import TYPE_CHECKING, Dict, List, Optional
+from typing import TYPE_CHECKING, Dict, List, Optional, Union
 
 from lightning_sdk.api.job_api import JobApiV1
 from lightning_sdk.api.utils import (
     _COMPUTE_NAME_TO_MACHINE,
-    _MACHINE_TO_COMPUTE_NAME,
     _create_app,
+    _machine_to_compute_name,
 )
 from lightning_sdk.api.utils import (
     _get_cloud_url as _cloud_url,
@@ -43,13 +43,13 @@ class MMTApiV1(JobApiV1):
         cloud_account: Optional[str],
         teamspace_id: str,
         studio_id: str,
-        machine: Machine,
+        machine: Union[Machine, str],
         interruptible: bool,
         strategy: str,
     ) -> Externalv1LightningappInstance:
         """Creates a multi-machine job with given commands."""
         distributed_args = {
-            "cloud_compute": _MACHINE_TO_COMPUTE_NAME[machine],
+            "cloud_compute": _machine_to_compute_name(machine),
             "num_instances": num_machines,
             "strategy": strategy,
         }
@@ -80,7 +80,7 @@ class MMTApiV2:
         teamspace_id: str,
         studio_id: Optional[str],
         image: Optional[str],
-        machine: Machine,
+        machine: Union[Machine, str],
         interruptible: bool,
         env: Optional[Dict[str, str]],
         image_credentials: Optional[str],
@@ -94,7 +94,7 @@ class MMTApiV2:
             for k, v in env.items():
                 env_vars.append(V1EnvVar(name=k, value=v))
 
-        instance_name = _MACHINE_TO_COMPUTE_NAME[machine]
+        instance_name = _machine_to_compute_name(machine)
 
         run_id = __GLOBAL_LIGHTNING_UNIQUE_IDS_STORE__[studio_id] if studio_id is not None else ""
 

lightning_sdk/cli/download.py

@@ -3,14 +3,18 @@ import re
 from pathlib import Path
 from typing import Optional
 
+from rich.console import Console
+
+from lightning_sdk.api.lit_container_api import LitContainerApi
 from lightning_sdk.cli.exceptions import StudioCliError
 from lightning_sdk.cli.studios_menu import _StudiosMenu
+from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
 from lightning_sdk.models import download_model
 from lightning_sdk.studio import Studio
 from lightning_sdk.utils.resolve import _get_authed_user, skip_studio_init
 
 
-class _Downloads(_StudiosMenu):
+class _Downloads(_StudiosMenu, _TeamspacesMenu):
     """Download files and folders from Lightning AI."""
 
     def model(self, name: str, download_dir: str = ".") -> None:
@@ -130,3 +134,18 @@ class _Downloads(_StudiosMenu):
                 f"Could not download the file from the given Studio {studio}. "
                 "Please contact Lightning AI directly to resolve this issue."
             ) from e
+
+    def container(self, container: str, teamspace: Optional[str] = None, tag: str = "latest") -> None:
+        """Download a docker container from a teamspace.
+
+        Args:
+          container: The name of the container to download.
+          teamspace: The name of the teamspace to download the container from.
+          tag: The tag of the container to download.
+        """
+        resolved_teamspace = self._resolve_teamspace(teamspace)
+        console = Console()
+        with console.status("Downloading container..."):
+            api = LitContainerApi()
+            api.download_container(container, resolved_teamspace, tag)
+            console.print("Container downloaded successfully", style="green")

lightning_sdk/cli/run.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Dict, Optional
+from typing import TYPE_CHECKING, Dict, Optional, Union
 
 from lightning_sdk.job import Job
 from lightning_sdk.machine import Machine
@@ -137,7 +137,11 @@ class _Run:
         if machine is None:
             # TODO: infer from studio
             machine = "CPU"
-        machine_enum = Machine[machine.upper()]
+        machine_enum: Union[str, Machine]
+        try:
+            machine_enum = Machine[machine.upper()]
+        except KeyError:
+            machine_enum = machine
 
         resolved_teamspace = Teamspace(name=teamspace, org=org, user=user)
 
@@ -195,7 +199,11 @@ class _Run:
         if machine is None:
             # TODO: infer from studio
             machine = "CPU"
-        machine_enum = Machine[machine.upper()]
+        machine_enum: Union[str, Machine]
+        try:
+            machine_enum = Machine[machine.upper()]
+        except KeyError:
+            machine_enum = machine
 
         resolved_teamspace = Teamspace(name=teamspace, org=org, user=user)
         if cloud_account is None:

lightning_sdk/job/base.py

@@ -16,7 +16,7 @@ if TYPE_CHECKING:
 class MachineDict(TypedDict):
     name: str
     status: "Status"
-    machine: "Machine"
+    machine: Union["Machine", str]
 
 
 class JobDict(MachineDict):
@@ -68,7 +68,7 @@ class _BaseJob(ABC):
     def run(
         cls,
         name: str,
-        machine: "Machine",
+        machine: Union["Machine", str],
         command: Optional[str] = None,
         studio: Union["Studio", str, None] = None,
         image: Optional[str] = None,
@@ -205,7 +205,7 @@ class _BaseJob(ABC):
     @abstractmethod
     def _submit(
         self,
-        machine: "Machine",
+        machine: Union["Machine", str],
         command: Optional[str] = None,
         studio: Optional["Studio"] = None,
         image: Optional[str] = None,
@@ -271,7 +271,7 @@ class _BaseJob(ABC):
 
     @property
     @abstractmethod
-    def machine(self) -> "Machine":
+    def machine(self) -> Union["Machine", str]:
         """The machine type the job is running on."""
 
     @property

lightning_sdk/job/job.py

@@ -89,7 +89,7 @@ class Job(_BaseJob):
     def run(
         cls,
         name: str,
-        machine: "Machine",
+        machine: Union["Machine", str],
         command: Optional[str] = None,
         studio: Union["Studio", str, None] = None,
         image: Union[str, None] = None,
@@ -169,7 +169,7 @@ class Job(_BaseJob):
 
     def _submit(
         self,
-        machine: "Machine",
+        machine: Union["Machine", str],
         command: Optional[str] = None,
         studio: Optional["Studio"] = None,
         image: Optional[str] = None,
@@ -225,6 +225,7 @@ class Job(_BaseJob):
             cloud_account_auth=cloud_account_auth,
             artifacts_local=artifacts_local,
             artifacts_remote=artifacts_remote,
+            entrypoint=entrypoint,
         )
         return self
 
@@ -248,7 +249,7 @@ class Job(_BaseJob):
         return self._internal_job.status
 
     @property
-    def machine(self) -> "Machine":
+    def machine(self) -> Union["Machine", str]:
         """The machine type the job is running on."""
         return self._internal_job.machine
 

lightning_sdk/job/v1.py

@@ -44,7 +44,7 @@ class _JobV1(_BaseJob):
     def run(
         cls,
         name: str,
-        machine: "Machine",
+        machine: Union["Machine", str],
         command: str,
         studio: "Studio",
         teamspace: Union[str, "Teamspace", None] = None,
@@ -89,7 +89,7 @@ class _JobV1(_BaseJob):
 
     def _submit(
         self,
-        machine: "Machine",
+        machine: Union["Machine", str],
         command: Optional[str] = None,
         studio: Optional["Studio"] = None,
         image: Optional[str] = None,
@@ -195,7 +195,7 @@ class _JobV1(_BaseJob):
         return Work(_work[0].id, self, self.teamspace)
 
     @property
-    def machine(self) -> "Machine":
+    def machine(self) -> Union["Machine", str]:
         """Get the machine the job is running on."""
         return self.work.machine
 

lightning_sdk/job/v2.py

@@ -37,7 +37,7 @@ class _JobV2(_BaseJob):
 
     def _submit(
         self,
-        machine: "Machine",
+        machine: Union["Machine", str],
         command: Optional[str] = None,
         studio: Optional["Studio"] = None,
         image: Optional[str] = None,
@@ -143,7 +143,7 @@ class _JobV2(_BaseJob):
         return self._job_api._job_state_to_external(self._latest_job.state)
 
     @property
-    def machine(self) -> "Machine":
+    def machine(self) -> Union["Machine", str]:
         """The machine type the job is running on."""
         # only fetch the job it it hasn't been fetched yet as machine cannot change over time
         return self._job_api._get_job_machine_from_spec(self._guaranteed_job.spec)

lightning_sdk/job/work.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Any, Optional, Protocol
+from typing import TYPE_CHECKING, Any, Optional, Protocol, Union
 
 from lightning_sdk.api.job_api import JobApiV1
 
@@ -51,7 +51,7 @@ class Work:
         return self._job._name_filter(self._guaranteed_work.name)
 
     @property
-    def machine(self) -> "Machine":
+    def machine(self) -> Union["Machine", str]:
         return self._job_api.get_machine_from_work(self._guaranteed_work)
 
     @property

lightning_sdk/lit_container.py

@@ -76,3 +76,22 @@ class LitContainer:
         resp = self._api.upload_container(container, teamspace, tag)
         for line in resp:
             print(line)
+
+    def download_container(
+        self, container: str, teamspace: str, org: Optional[str] = None, user: Optional[str] = None, tag: str = "latest"
+    ) -> None:
+        """Download a container from the docker registry.
+
+        Args:
+            container: The name of the container to download.
+            teamspace: The teamspace which contains the container.
+            org: The organization which contains the container.
+            user: The user which contains the container.
+            tag: The tag to use for the container.
+        """
+        try:
+            teamspace = _resolve_teamspace(teamspace=teamspace, org=org, user=user)
+        except Exception as e:
+            raise ValueError(f"Could not resolve teamspace: {e}") from e
+
+        return self._api.download_container(container, teamspace, tag)

lightning_sdk/mmt/base.py

@@ -23,7 +23,7 @@ class MMTMachine(Protocol):
         ...
 
     @property
-    def machine(self) -> "Machine":
+    def machine(self) -> Union["Machine", str]:
         """The actual machine type this node is running on."""
         ...
 
@@ -54,7 +54,7 @@ class _BaseMMT(_BaseJob):
     def run(
         cls,
         name: str,
-        machine: "Machine",
+        machine: Union["Machine", str],
         num_machines: int,
         command: Optional[str] = None,
         studio: Union["Studio", str, None] = None,
@@ -199,7 +199,7 @@ class _BaseMMT(_BaseJob):
     def _submit(
         self,
         num_machines: int,
-        machine: "Machine",
+        machine: Union["Machine", str],
         command: Optional[str] = None,
         studio: Optional["Studio"] = None,
         image: Optional[str] = None,
@@ -257,7 +257,7 @@ class _BaseMMT(_BaseJob):
 
     @property
     @abstractmethod
-    def machine(self) -> "Machine":
+    def machine(self) -> Union["Machine", str]:
         """Returns the machine type this job is running on."""
 
     @abstractmethod

lightning_sdk/mmt/mmt.py

@@ -97,7 +97,7 @@ class MMT(_BaseMMT):
         cls,
         name: str,
         num_machines: int,
-        machine: "Machine",
+        machine: Union["Machine", str],
         command: Optional[str] = None,
         studio: Union["Studio", str, None] = None,
         image: Union[str, None] = None,
@@ -167,6 +167,7 @@ class MMT(_BaseMMT):
             cloud_account_auth=cloud_account_auth,
             artifacts_local=artifacts_local,
             artifacts_remote=artifacts_remote,
+            entrypoint=entrypoint,
             cluster=cluster,  # deprecated in favor of cloud_account
         )
         # required for typing with "MMT"
@@ -180,7 +181,7 @@ class MMT(_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,
@@ -239,6 +240,7 @@ class MMT(_BaseMMT):
             cloud_account_auth=cloud_account_auth,
             artifacts_local=artifacts_local,
             artifacts_remote=artifacts_remote,
+            entrypoint=entrypoint,
         )
         return self
 
@@ -264,7 +266,7 @@ class MMT(_BaseMMT):
         return self._internal_mmt.machines
 
     @property
-    def machine(self) -> "Machine":
+    def machine(self) -> Union["Machine", str]:
         """Returns the machine type this job is running on."""
         return self._internal_mmt.machine
 

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,
@@ -166,7 +166,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,
@@ -169,7 +169,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-0.1.49.dist-info → lightning_sdk-0.1.50.dist-info}/METADATA

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

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

@@ -1,10 +1,10 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
-lightning_sdk/__init__.py,sha256=pghaQ6_YR2SmOR4N7VM5mlRGqecB5VgWJpqV0JMLxiE,970
+lightning_sdk/__init__.py,sha256=7m3PnW4ypq_Vcr5POTDAdFl_IzMe1ouPzuhbPqkGbiU,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/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/organization.py,sha256=WCfzdgjtvY1_A07DnxOpp74V2JR2gQwtXbIEcFDnoVU,1232
@@ -18,9 +18,9 @@ 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=C9-6mwy5ufxgK7Q8YhMa8U90B_bNIbA8yo-aLvAp1w8,12261
+lightning_sdk/api/lit_container_api.py,sha256=yuzALzKHd_vwFNaPbxmPvLWAjItB1npKnVj1QMJbXl8,2541
+lightning_sdk/api/mmt_api.py,sha256=zmNmoSX6JPRQiHlBcGVGdk2LGYzlwiOyP0Io2NNyWDU,7353
 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
@@ -29,7 +29,7 @@ lightning_sdk/api/utils.py,sha256=VG3sCxWjn6MHONYghA73PZH-ErgqsHyj1ZMU6qzBCFk,22
 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,7 +38,7 @@ 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/run.py,sha256=80fZInp90YyfSRdEQZbNaebZ7jQWbusNYn0FTWKH9c4,12051
 lightning_sdk/cli/serve.py,sha256=UaXhGHU6nbAzrnVigSKOTrMjLwSs-sjyhuJCdVUBwzc,8722
 lightning_sdk/cli/stop.py,sha256=qBlVaYAJGxiNSwskxqB7zizTVPIpBGlJ4c7kIXII7iw,1373
 lightning_sdk/cli/studios_menu.py,sha256=0kQGqGel8gAbpdJtjOM1a6NEat_TnIqRNprNn8QiK58,3236
@@ -47,11 +47,11 @@ lightning_sdk/cli/upload.py,sha256=pn7yIq98e_gMoaFhETHfiXai_15qdyG6t1hsdjN-oWg,1
 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=xQ6zDUmCSncQ1mqSQP--VNOjyNxtiSjAykKC1_aFG1s,16320
+lightning_sdk/job/job.py,sha256=Ikq0T9exdsYS77OEsQFoxMjKCgYfVdXMOlbBaZFCZ1w,13496
+lightning_sdk/job/v1.py,sha256=CYiKiIFGEBmns0iDRybalp_1TRBGEfvu5lCJLvyvq84,10238
+lightning_sdk/job/v2.py,sha256=EGkFuNxtixR7h1q0ld4Lm1OQk5fG4ox5jqsIO7ZAGIk,10204
+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
@@ -851,10 +851,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=1w91ggRWAAvUmBAnVqY2ecHU_lLz0_7AhJrtR1b-PPs,14870
+lightning_sdk/mmt/mmt.py,sha256=lkrg-LYHyQLFIoV4A9C5BvjUFAefLeHzgeDbmpF1NEs,14135
+lightning_sdk/mmt/v1.py,sha256=SHKQZYG3enK8idyO96S54DnRdEej4n38TKHBO8YEeZo,9371
+lightning_sdk/mmt/v2.py,sha256=_qDGWaeHoW3laJ_cSglNN_JTGSxAiGeaZfQIX6iGLG8,9706
 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
@@ -863,9 +863,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.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-0.1.50.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-0.1.50.dist-info/METADATA,sha256=XLrPyTcYOTNrqvE8jTtECxvaJ1Ev19fwe2Qk4BMaJ6E,4031
+lightning_sdk-0.1.50.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+lightning_sdk-0.1.50.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
+lightning_sdk-0.1.50.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-0.1.50.dist-info/RECORD,,