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

lightning_sdk/__init__.py

@@ -4,6 +4,7 @@ from lightning_sdk.constants import __GLOBAL_LIGHTNING_UNIQUE_IDS_STORE__  # noq
 from lightning_sdk.helpers import _check_version_and_prompt_upgrade
 from lightning_sdk.job import Job
 from lightning_sdk.machine import Machine
+from lightning_sdk.mmt import MMT
 from lightning_sdk.organization import Organization
 from lightning_sdk.plugin import JobsPlugin, MultiMachineTrainingPlugin, Plugin, SlurmJobsPlugin
 from lightning_sdk.status import Status
@@ -15,6 +16,7 @@ __all__ = [
     "Job",
     "JobsPlugin",
     "Machine",
+    "MMT",
     "MultiMachineTrainingPlugin",
     "Organization",
     "Plugin",
@@ -27,5 +29,5 @@ __all__ = [
     "AIHub",
 ]
 
-__version__ = "0.1.48"
+__version__ = "0.1.49"
 _check_version_and_prompt_upgrade(__version__)

lightning_sdk/api/job_api.py

@@ -180,6 +180,10 @@ class JobApiV1:
 
         raise RuntimeError("Could not extract command from app")
 
+    def get_total_cost(self, job: Externalv1LightningappInstance) -> float:
+        status: V1LightningappInstanceStatus = job.status
+        return status.total_cost
+
 
 class JobApiV2:
     # these are stages the job can be in.
@@ -337,3 +341,6 @@ class JobApiV2:
         return _COMPUTE_NAME_TO_MACHINE.get(
             instance_type, _COMPUTE_NAME_TO_MACHINE.get(instance_name, instance_type or instance_name)
         )
+
+    def get_total_cost(self, job: V1Job) -> float:
+        return job.total_cost

lightning_sdk/api/lit_container_api.py

@@ -1,7 +1,9 @@
-from typing import List
+from typing import Generator, List
 
+from lightning_sdk.api.utils import _get_registry_url
 from lightning_sdk.lightning_cloud.openapi.models import V1DeleteLitRepositoryResponse
 from lightning_sdk.lightning_cloud.rest_client import LightningClient
+from lightning_sdk.teamspace import Teamspace
 
 
 class LitContainerApi:
@@ -17,3 +19,24 @@ class LitContainerApi:
             return self._client.lit_registry_service_delete_lit_repository(project_id, container)
         except Exception as ex:
             raise ValueError(f"Could not delete container {container} from project {project_id}") from ex
+
+    def upload_container(self, container: str, teamspace: Teamspace, tag: str) -> Generator[str, None, None]:
+        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)
+        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)
+        if not tagged:
+            raise ValueError(f"Could not tag container {container} with {repository}:{tag}")
+        return client.api.push(repository, stream=True, decode=True)

lightning_sdk/api/mmt_api.py

@@ -203,3 +203,9 @@ class MMTApiV2:
         return _COMPUTE_NAME_TO_MACHINE.get(
             instance_type, _COMPUTE_NAME_TO_MACHINE.get(instance_name, instance_type or instance_name)
         )
+
+    def get_total_cost(self, job: V1MultiMachineJob) -> float:
+        return job.total_cost
+
+    def get_num_machines(self, job: V1MultiMachineJob) -> int:
+        return job.machines

lightning_sdk/api/utils.py

@@ -354,6 +354,7 @@ def _machine_to_compute_name(machine: Union[Machine, str]) -> str:
 _COMPUTE_NAME_TO_MACHINE: Dict[str, Machine] = {v: k for k, v in _MACHINE_TO_COMPUTE_NAME.items()}
 
 _DEFAULT_CLOUD_URL = "https://lightning.ai"
+_DEFAULT_REGISTRY_URL = "litcr.io"
 
 
 def _get_cloud_url() -> str:
@@ -362,6 +363,12 @@ def _get_cloud_url() -> str:
     return cloud_url
 
 
+def _get_registry_url() -> str:
+    registry_url = os.environ.get("LIGHTNING_REGISTRY_URL", _DEFAULT_REGISTRY_URL)
+    os.environ["LIGHTNING_REGISTRY_URL"] = registry_url
+    return registry_url
+
+
 def _sanitize_studio_remote_path(path: str, studio_id: str) -> str:
     return f"/cloudspaces/{studio_id}/code/content/{path.replace('/teamspace/studios/this_studio/', '')}"
 

lightning_sdk/cli/entrypoint.py

@@ -1,3 +1,7 @@
+import sys
+from types import TracebackType
+from typing import Type
+
 from fire import Fire
 from lightning_utilities.core.imports import RequirementCache
 
@@ -32,6 +36,8 @@ class StudioCLI:
         self.inspect = _Inspect()
         self.stop = _Stop()
 
+        sys.excepthook = _notify_exception
+
     def login(self) -> None:
         """Login to Lightning AI Studios."""
         auth = Auth()
@@ -48,6 +54,11 @@ class StudioCLI:
         auth.clear()
 
 
+def _notify_exception(exception_type: Type[BaseException], value: BaseException, tb: TracebackType) -> None:  # No
+    """CLI won't show tracebacks, just print the exception message."""
+    print(value)
+
+
 def main_cli() -> None:
     """CLI entrypoint."""
     Fire(StudioCLI(), name="lightning")

lightning_sdk/cli/list.py

@@ -20,7 +20,36 @@ class _List(_TeamspacesMenu):
         """
         resolved_teamspace = self._resolve_teamspace(teamspace=teamspace)
 
-        print("Available Jobs:\n" + "\n".join([j.name for j in resolved_teamspace.jobs]))
+        jobs = resolved_teamspace.jobs
+
+        table = Table(
+            pad_edge=True,
+        )
+        table.add_column("Name")
+        table.add_column("Teamspace")
+        table.add_column("Studio")
+        table.add_column("Image")
+        table.add_column("Status")
+        table.add_column("Machine")
+        table.add_column("Total Cost")
+        for j in jobs:
+            # we know we just fetched these, so no need to refetch
+            j._prevent_refetch_latest = True
+            j._internal_job._prevent_refetch_latest = True
+
+            studio = j.studio
+            table.add_row(
+                j.name,
+                f"{j.teamspace.owner.name}/{j.teamspace.name}",
+                studio.name if studio else None,
+                j.image,
+                str(j.status),
+                str(j.machine),
+                f"{j.total_cost:.3f}",
+            )
+
+        console = Console()
+        console.print(table)
 
     def mmts(self, teamspace: Optional[str] = None) -> None:
         """List multi-machine jobs for a given teamspace.
@@ -32,7 +61,36 @@ class _List(_TeamspacesMenu):
         """
         resolved_teamspace = self._resolve_teamspace(teamspace=teamspace)
 
-        print("Available MMTs:\n" + "\n".join([j.name for j in resolved_teamspace.multi_machine_jobs]))
+        jobs = resolved_teamspace.multi_machine_jobs
+
+        table = Table(pad_edge=True)
+        table.add_column("Name")
+        table.add_column("Teamspace")
+        table.add_column("Studio")
+        table.add_column("Image")
+        table.add_column("Status")
+        table.add_column("Machine")
+        table.add_column("Num Machines")
+        table.add_column("Total Cost")
+        for j in jobs:
+            # we know we just fetched these, so no need to refetch
+            j._prevent_refetch_latest = True
+            j._internal_job._prevent_refetch_latest = True
+
+            studio = j.studio
+            table.add_row(
+                j.name,
+                f"{j.teamspace.owner.name}/{j.teamspace.name}",
+                studio.name if studio else None,
+                j.image,
+                str(j.status),
+                str(j.machine),
+                str(j.num_machines),
+                str(j.total_cost),
+            )
+
+        console = Console()
+        console.print(table)
 
     def containers(self, teamspace: Optional[str] = None) -> None:
         """Display the list of available containers.

lightning_sdk/cli/run.py

@@ -111,7 +111,7 @@ class _Run:
     # might need to move to different cli library
     def job(
         self,
-        name: str,
+        name: Optional[str] = None,
         machine: Optional[str] = None,
         command: Optional[str] = None,
         studio: Optional[str] = None,
@@ -128,16 +128,23 @@ class _Run:
         artifacts_remote: Optional[str] = None,
         entrypoint: str = "sh -c",
     ) -> None:
+        if not name:
+            from datetime import datetime
+
+            timestr = datetime.now().strftime("%Y-%m-%d-%H-%M-%S")
+            name = f"job-{timestr}"
+
         if machine is None:
             # TODO: infer from studio
             machine = "CPU"
-        machine_enum = Machine(machine.upper())
+        machine_enum = Machine[machine.upper()]
 
         resolved_teamspace = Teamspace(name=teamspace, org=org, user=user)
 
         if cloud_account is None:
             cloud_account = resolved_teamspace.default_cloud_account
         machine_enum = Machine(machine.upper())
+
         Job.run(
             name=name,
             machine=machine_enum,
@@ -188,7 +195,7 @@ class _Run:
         if machine is None:
             # TODO: infer from studio
             machine = "CPU"
-        machine_enum = Machine(machine.upper())
+        machine_enum = Machine[machine.upper()]
 
         resolved_teamspace = Teamspace(name=teamspace, org=org, user=user)
         if cloud_account is None:

lightning_sdk/cli/upload.py

@@ -4,18 +4,22 @@ import os
 from pathlib import Path
 from typing import Dict, List, Optional
 
+from rich.console import Console
+from rich.progress import Progress, SpinnerColumn, TextColumn, TimeElapsedColumn
 from simple_term_menu import TerminalMenu
 from tqdm import tqdm
 
+from lightning_sdk.api.lit_container_api import LitContainerApi
 from lightning_sdk.api.utils import _get_cloud_url
 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 upload_model
 from lightning_sdk.studio import Studio
 from lightning_sdk.utils.resolve import _get_authed_user, skip_studio_init
 
 
-class _Uploads(_StudiosMenu):
+class _Uploads(_StudiosMenu, _TeamspacesMenu):
     """Upload files and folders to Lightning AI."""
 
     _studio_upload_status_path = "~/.lightning/studios/uploads"
@@ -146,6 +150,33 @@ class _Uploads(_StudiosMenu):
         )
         print(f"See your file at {studio_url}")
 
+    def container(self, container: str, tag: str = "latest", teamspace: Optional[str] = None) -> None:
+        teamspace = self._resolve_teamspace(teamspace)
+        api = LitContainerApi()
+        console = Console()
+        with Progress(
+            SpinnerColumn(),
+            TextColumn("[progress.description]{task.description}"),
+            TimeElapsedColumn(),
+            console=console,
+            transient=False,
+        ) as progress:
+            push_task = progress.add_task("Pushing Docker image", total=None)
+            resp = api.upload_container(container, teamspace, tag)
+            for line in resp:
+                if "status" in line:
+                    console.print(line["status"], style="bright_black")
+                    progress.update(push_task, description="Pushing Docker image")
+                elif "aux" in line:
+                    console.print(line["aux"], style="bright_black")
+                elif "error" in line:
+                    progress.stop()
+                    console.print(f"\n[red]{line}[/red]")
+                    return
+                else:
+                    console.print(line, style="bright_black")
+            progress.update(push_task, description="[green]Container pushed![/green]")
+
     def _start_parallel_upload(
         self, executor: concurrent.futures.ThreadPoolExecutor, studio: Studio, upload_state: Dict[str, str]
     ) -> List[concurrent.futures.Future]:

lightning_sdk/job/base.py

@@ -24,6 +24,7 @@ class JobDict(MachineDict):
     teamspace: str
     studio: Optional[str]
     image: Optional[str]
+    total_cost: float
 
 
 class _BaseJob(ABC):
@@ -61,6 +62,8 @@ class _BaseJob(ABC):
         if _fetch_job:
             self._update_internal_job()
 
+        self._prevent_refetch_latest = False
+
     @classmethod
     def run(
         cls,
@@ -332,6 +335,7 @@ class _BaseJob(ABC):
             "command": self.command,
             "status": self.status,
             "machine": self.machine,
+            "total_cost": self.total_cost,
         }
 
     def json(self) -> str:
@@ -361,3 +365,18 @@ class _BaseJob(ABC):
             self._update_internal_job()
 
         return self._job
+
+    @property
+    def total_cost(self) -> float:
+        """The number of credits the job was consuming so far."""
+        return self._job_api.get_total_cost(self._latest_job)
+
+    @property
+    def _latest_job(self) -> Any:
+        """Guarantees to fetch the latest version of a job before returning it."""
+        # in some cases we know we just refetched the latest state, no need to refetch again
+        if self._prevent_refetch_latest:
+            return self._guaranteed_job
+
+        self._update_internal_job()
+        return self._job

lightning_sdk/job/v1.py

@@ -174,7 +174,7 @@ class _JobV1(_BaseJob):
 
     def stop(self) -> None:
         """Stops the job. is blocking until the ob is stopped."""
-        if self.status in (Status.Stopped, Status.Failed):
+        if self.status in (Status.Stopped, Status.Completed, Status.Failed):
             return None
 
         return self._job_api.stop_job(self._job.id, self.teamspace.id)

lightning_sdk/job/v2.py

@@ -1,13 +1,13 @@
-from typing import TYPE_CHECKING, Any, Dict, Optional, Union
+from typing import TYPE_CHECKING, Dict, Optional, Union
 
 from lightning_sdk.api.job_api import JobApiV2
 from lightning_sdk.api.utils import _get_cloud_url
 from lightning_sdk.job.base import _BaseJob
+from lightning_sdk.status import Status
 
 if TYPE_CHECKING:
     from lightning_sdk.machine import Machine
     from lightning_sdk.organization import Organization
-    from lightning_sdk.status import Status
     from lightning_sdk.studio import Studio
     from lightning_sdk.teamspace import Teamspace
     from lightning_sdk.user import User
@@ -121,6 +121,9 @@ class _JobV2(_BaseJob):
 
     def stop(self) -> None:
         """Stop the job. If the job is already stopped, this is a no-op. This is blocking until the job is stopped."""
+        if self.status in (Status.Stopped, Status.Completed, Status.Failed):
+            return
+
         self._job_api.stop_job(job_id=self._guaranteed_job.id, teamspace_id=self._teamspace.id)
 
     def delete(self) -> None:
@@ -134,12 +137,6 @@ class _JobV2(_BaseJob):
             cloudspace_id=self._guaranteed_job.spec.cloudspace_id,
         )
 
-    @property
-    def _latest_job(self) -> Any:
-        """Guarantees to fetch the latest version of a job before returning it."""
-        self._update_internal_job()
-        return self._job
-
     @property
     def status(self) -> "Status":
         """The current status of the job."""

lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py

@@ -42,7 +42,6 @@ class V1ClusterSpec(object):
     """
     swagger_types = {
         'auth_token': 'str',
-        'available_instance_types': 'list[str]',
         'aws_v1': 'V1AWSDirectV1',
         'cluster_type': 'V1ClusterType',
         'deletion_options': 'V1ClusterDeletionOptions',
@@ -63,7 +62,6 @@ class V1ClusterSpec(object):
 
     attribute_map = {
         'auth_token': 'authToken',
-        'available_instance_types': 'availableInstanceTypes',
         'aws_v1': 'awsV1',
         'cluster_type': 'clusterType',
         'deletion_options': 'deletionOptions',
@@ -82,10 +80,9 @@ class V1ClusterSpec(object):
         'vultr_v1': 'vultrV1'
     }
 
-    def __init__(self, auth_token: 'str' =None, available_instance_types: 'list[str]' =None, aws_v1: 'V1AWSDirectV1' =None, cluster_type: 'V1ClusterType' =None, deletion_options: 'V1ClusterDeletionOptions' =None, desired_state: 'V1ClusterState' =None, domain: 'str' =None, freeze_accelerators: 'bool' =None, google_cloud_v1: 'V1GoogleCloudDirectV1' =None, insurer_disabled: 'bool' =None, lambda_labs_v1: 'V1LambdaLabsDirectV1' =None, overprovisioning: 'list[V1InstanceOverprovisioningSpec]' =None, pause_automation: 'bool' =None, security_options: 'V1ClusterSecurityOptions' =None, slurm_v1: 'V1SlurmV1' =None, tagging_options: 'V1ClusterTaggingOptions' =None, user_id: 'str' =None, vultr_v1: 'V1VultrDirectV1' =None):  # noqa: E501
+    def __init__(self, auth_token: 'str' =None, aws_v1: 'V1AWSDirectV1' =None, cluster_type: 'V1ClusterType' =None, deletion_options: 'V1ClusterDeletionOptions' =None, desired_state: 'V1ClusterState' =None, domain: 'str' =None, freeze_accelerators: 'bool' =None, google_cloud_v1: 'V1GoogleCloudDirectV1' =None, insurer_disabled: 'bool' =None, lambda_labs_v1: 'V1LambdaLabsDirectV1' =None, overprovisioning: 'list[V1InstanceOverprovisioningSpec]' =None, pause_automation: 'bool' =None, security_options: 'V1ClusterSecurityOptions' =None, slurm_v1: 'V1SlurmV1' =None, tagging_options: 'V1ClusterTaggingOptions' =None, user_id: 'str' =None, vultr_v1: 'V1VultrDirectV1' =None):  # noqa: E501
         """V1ClusterSpec - a model defined in Swagger"""  # noqa: E501
         self._auth_token = None
-        self._available_instance_types = None
         self._aws_v1 = None
         self._cluster_type = None
         self._deletion_options = None
@@ -105,8 +102,6 @@ class V1ClusterSpec(object):
         self.discriminator = None
         if auth_token is not None:
             self.auth_token = auth_token
-        if available_instance_types is not None:
-            self.available_instance_types = available_instance_types
         if aws_v1 is not None:
             self.aws_v1 = aws_v1
         if cluster_type is not None:
@@ -161,29 +156,6 @@ class V1ClusterSpec(object):
 
         self._auth_token = auth_token
 
-    @property
-    def available_instance_types(self) -> 'list[str]':
-        """Gets the available_instance_types of this V1ClusterSpec.  # noqa: E501
-
-        available_instance_types is a list of instance types that are available for the cluster. This is just a soft filter to prevent users from using instances that we haven't prepared for. If the list is empty, no filtering is done.  # noqa: E501
-
-        :return: The available_instance_types of this V1ClusterSpec.  # noqa: E501
-        :rtype: list[str]
-        """
-        return self._available_instance_types
-
-    @available_instance_types.setter
-    def available_instance_types(self, available_instance_types: 'list[str]'):
-        """Sets the available_instance_types of this V1ClusterSpec.
-
-        available_instance_types is a list of instance types that are available for the cluster. This is just a soft filter to prevent users from using instances that we haven't prepared for. If the list is empty, no filtering is done.  # noqa: E501
-
-        :param available_instance_types: The available_instance_types of this V1ClusterSpec.  # noqa: E501
-        :type: list[str]
-        """
-
-        self._available_instance_types = available_instance_types
-
     @property
     def aws_v1(self) -> 'V1AWSDirectV1':
         """Gets the aws_v1 of this V1ClusterSpec.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_lambda_labs_direct_v1.py

@@ -41,19 +41,24 @@ class V1LambdaLabsDirectV1(object):
                             and the value is json key in definition.
     """
     swagger_types = {
-        'credentials_secret_id': 'str'
+        'credentials_secret_id': 'str',
+        'parent_cluster_id': 'str'
     }
 
     attribute_map = {
-        'credentials_secret_id': 'credentialsSecretId'
+        'credentials_secret_id': 'credentialsSecretId',
+        'parent_cluster_id': 'parentClusterId'
     }
 
-    def __init__(self, credentials_secret_id: 'str' =None):  # noqa: E501
+    def __init__(self, credentials_secret_id: 'str' =None, parent_cluster_id: 'str' =None):  # noqa: E501
         """V1LambdaLabsDirectV1 - a model defined in Swagger"""  # noqa: E501
         self._credentials_secret_id = None
+        self._parent_cluster_id = None
         self.discriminator = None
         if credentials_secret_id is not None:
             self.credentials_secret_id = credentials_secret_id
+        if parent_cluster_id is not None:
+            self.parent_cluster_id = parent_cluster_id
 
     @property
     def credentials_secret_id(self) -> 'str':
@@ -78,6 +83,29 @@ class V1LambdaLabsDirectV1(object):
 
         self._credentials_secret_id = credentials_secret_id
 
+    @property
+    def parent_cluster_id(self) -> 'str':
+        """Gets the parent_cluster_id of this V1LambdaLabsDirectV1.  # noqa: E501
+
+        Note: LambdaLabs is missing object store in their offering, we will need to use either GCP or AWS S3 for that.  # noqa: E501
+
+        :return: The parent_cluster_id of this V1LambdaLabsDirectV1.  # noqa: E501
+        :rtype: str
+        """
+        return self._parent_cluster_id
+
+    @parent_cluster_id.setter
+    def parent_cluster_id(self, parent_cluster_id: 'str'):
+        """Sets the parent_cluster_id of this V1LambdaLabsDirectV1.
+
+        Note: LambdaLabs is missing object store in their offering, we will need to use either GCP or AWS S3 for that.  # noqa: E501
+
+        :param parent_cluster_id: The parent_cluster_id of this V1LambdaLabsDirectV1.  # noqa: E501
+        :type: str
+        """
+
+        self._parent_cluster_id = parent_cluster_id
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py

@@ -120,6 +120,7 @@ class V1UserFeatures(object):
         'teamspace_storage_tab': 'bool',
         'trainium2': 'bool',
         'use_rclone_mounts_only': 'bool',
+        'vultr': 'bool',
         'writable_data_connections': 'bool'
     }
 
@@ -203,10 +204,11 @@ class V1UserFeatures(object):
         'teamspace_storage_tab': 'teamspaceStorageTab',
         'trainium2': 'trainium2',
         'use_rclone_mounts_only': 'useRcloneMountsOnly',
+        'vultr': 'vultr',
         'writable_data_connections': 'writableDataConnections'
     }
 
-    def __init__(self, advanced_deployment_autoscaling: 'bool' =None, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, b2c_experience: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, custom_instance_types: 'bool' =None, default_one_cluster: 'bool' =None, deployment_customize_api: 'bool' =None, deployment_data_path: 'bool' =None, deployment_gallery: 'bool' =None, deployment_persistent_disk: 'bool' =None, deployment_version_visibility: 'bool' =None, docs_agent: 'bool' =None, drive_v2: 'bool' =None, enable_crypto_crackdown: 'bool' =None, enable_efs: 'bool' =None, enable_storage_limits: 'bool' =None, featured_studios_admin: 'bool' =None, filesystem_optimisation: 'bool' =None, gcp: 'bool' =None, inference_job_deployment_plugin: 'bool' =None, instant_capacity_reservation: 'bool' =None, jobs_init: 'bool' =None, jobs_v2: 'bool' =None, landing_studios: 'bool' =None, lightning_registry: 'bool' =None, lit_logger: 'bool' =None, lit_logger_storage_v2: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, mmt_v2: 'bool' =None, model_store: 'bool' =None, multiple_deployment_versions: 'bool' =None, multiple_studio_versions: 'bool' =None, org_level_member_permissions: 'bool' =None, pipelines: 'bool' =None, plugin_biz_chat: 'bool' =None, plugin_distributed: 'bool' =None, plugin_fiftyone: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_lightning_apps: 'bool' =None, plugin_lightning_apps_distributed: 'bool' =None, plugin_mage_ai: 'bool' =None, plugin_milvus: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_react: 'bool' =None, plugin_service: 'bool' =None, plugin_sweeps: 'bool' =None, plugin_weviate: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, project_selector: 'bool' =None, restart_ide_on_hang: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, show_dev_admin: 'bool' =None, slurm: 'bool' =None, slurm_machine_selector: 'bool' =None, snapshotter_service: 'bool' =None, snowflake_connection: 'bool' =None, spot_v2: 'bool' =None, studio_config: 'bool' =None, studio_on_stop: 'bool' =None, studio_version_visibility: 'bool' =None, teamspace_storage_tab: 'bool' =None, trainium2: 'bool' =None, use_rclone_mounts_only: 'bool' =None, writable_data_connections: 'bool' =None):  # noqa: E501
+    def __init__(self, advanced_deployment_autoscaling: 'bool' =None, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, b2c_experience: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, custom_instance_types: 'bool' =None, default_one_cluster: 'bool' =None, deployment_customize_api: 'bool' =None, deployment_data_path: 'bool' =None, deployment_gallery: 'bool' =None, deployment_persistent_disk: 'bool' =None, deployment_version_visibility: 'bool' =None, docs_agent: 'bool' =None, drive_v2: 'bool' =None, enable_crypto_crackdown: 'bool' =None, enable_efs: 'bool' =None, enable_storage_limits: 'bool' =None, featured_studios_admin: 'bool' =None, filesystem_optimisation: 'bool' =None, gcp: 'bool' =None, inference_job_deployment_plugin: 'bool' =None, instant_capacity_reservation: 'bool' =None, jobs_init: 'bool' =None, jobs_v2: 'bool' =None, landing_studios: 'bool' =None, lightning_registry: 'bool' =None, lit_logger: 'bool' =None, lit_logger_storage_v2: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, mmt_v2: 'bool' =None, model_store: 'bool' =None, multiple_deployment_versions: 'bool' =None, multiple_studio_versions: 'bool' =None, org_level_member_permissions: 'bool' =None, pipelines: 'bool' =None, plugin_biz_chat: 'bool' =None, plugin_distributed: 'bool' =None, plugin_fiftyone: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_lightning_apps: 'bool' =None, plugin_lightning_apps_distributed: 'bool' =None, plugin_mage_ai: 'bool' =None, plugin_milvus: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_react: 'bool' =None, plugin_service: 'bool' =None, plugin_sweeps: 'bool' =None, plugin_weviate: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, project_selector: 'bool' =None, restart_ide_on_hang: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, show_dev_admin: 'bool' =None, slurm: 'bool' =None, slurm_machine_selector: 'bool' =None, snapshotter_service: 'bool' =None, snowflake_connection: 'bool' =None, spot_v2: 'bool' =None, studio_config: 'bool' =None, studio_on_stop: 'bool' =None, studio_version_visibility: 'bool' =None, teamspace_storage_tab: 'bool' =None, trainium2: 'bool' =None, use_rclone_mounts_only: 'bool' =None, vultr: 'bool' =None, writable_data_connections: 'bool' =None):  # noqa: E501
         """V1UserFeatures - a model defined in Swagger"""  # noqa: E501
         self._advanced_deployment_autoscaling = None
         self._affiliate_links = None
@@ -287,6 +289,7 @@ class V1UserFeatures(object):
         self._teamspace_storage_tab = None
         self._trainium2 = None
         self._use_rclone_mounts_only = None
+        self._vultr = None
         self._writable_data_connections = None
         self.discriminator = None
         if advanced_deployment_autoscaling is not None:
@@ -447,6 +450,8 @@ class V1UserFeatures(object):
             self.trainium2 = trainium2
         if use_rclone_mounts_only is not None:
             self.use_rclone_mounts_only = use_rclone_mounts_only
+        if vultr is not None:
+            self.vultr = vultr
         if writable_data_connections is not None:
             self.writable_data_connections = writable_data_connections
 
@@ -2109,6 +2114,27 @@ class V1UserFeatures(object):
 
         self._use_rclone_mounts_only = use_rclone_mounts_only
 
+    @property
+    def vultr(self) -> 'bool':
+        """Gets the vultr of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The vultr of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._vultr
+
+    @vultr.setter
+    def vultr(self, vultr: 'bool'):
+        """Sets the vultr of this V1UserFeatures.
+
+
+        :param vultr: The vultr of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._vultr = vultr
+
     @property
     def writable_data_connections(self) -> 'bool':
         """Gets the writable_data_connections of this V1UserFeatures.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_vultr_direct_v1.py

@@ -42,24 +42,29 @@ class V1VultrDirectV1(object):
     """
     swagger_types = {
         'credentials_secret_id': 'str',
+        'parent_cluster_id': 'str',
         'primary_region': 'str',
         'regions': 'list[str]'
     }
 
     attribute_map = {
         'credentials_secret_id': 'credentialsSecretId',
+        'parent_cluster_id': 'parentClusterId',
         'primary_region': 'primaryRegion',
         'regions': 'regions'
     }
 
-    def __init__(self, credentials_secret_id: 'str' =None, primary_region: 'str' =None, regions: 'list[str]' =None):  # noqa: E501
+    def __init__(self, credentials_secret_id: 'str' =None, parent_cluster_id: 'str' =None, primary_region: 'str' =None, regions: 'list[str]' =None):  # noqa: E501
         """V1VultrDirectV1 - a model defined in Swagger"""  # noqa: E501
         self._credentials_secret_id = None
+        self._parent_cluster_id = None
         self._primary_region = None
         self._regions = None
         self.discriminator = None
         if credentials_secret_id is not None:
             self.credentials_secret_id = credentials_secret_id
+        if parent_cluster_id is not None:
+            self.parent_cluster_id = parent_cluster_id
         if primary_region is not None:
             self.primary_region = primary_region
         if regions is not None:
@@ -88,6 +93,27 @@ class V1VultrDirectV1(object):
 
         self._credentials_secret_id = credentials_secret_id
 
+    @property
+    def parent_cluster_id(self) -> 'str':
+        """Gets the parent_cluster_id of this V1VultrDirectV1.  # noqa: E501
+
+
+        :return: The parent_cluster_id of this V1VultrDirectV1.  # noqa: E501
+        :rtype: str
+        """
+        return self._parent_cluster_id
+
+    @parent_cluster_id.setter
+    def parent_cluster_id(self, parent_cluster_id: 'str'):
+        """Sets the parent_cluster_id of this V1VultrDirectV1.
+
+
+        :param parent_cluster_id: The parent_cluster_id of this V1VultrDirectV1.  # noqa: E501
+        :type: str
+        """
+
+        self._parent_cluster_id = parent_cluster_id
+
     @property
     def primary_region(self) -> 'str':
         """Gets the primary_region of this V1VultrDirectV1.  # noqa: E501

lightning_sdk/lit_container.py

@@ -55,3 +55,24 @@ class LitContainer:
             raise ValueError("Could not resolve teamspace") from e
         project_id = teamspace.id
         return self._api.delete_container(project_id, container)
+
+    def upload_container(
+        self, container: str, teamspace: str, org: Optional[str] = None, user: Optional[str] = None, tag: str = "latest"
+    ) -> None:
+        """Upload a container to the docker registry.
+
+        Args:
+            container: The name of the container to upload.
+            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
+
+        resp = self._api.upload_container(container, teamspace, tag)
+        for line in resp:
+            print(line)

lightning_sdk/mmt/base.py

@@ -250,6 +250,11 @@ class _BaseMMT(_BaseJob):
     def machines(self) -> Tuple[MMTMachine, ...]:
         """Returns the sub-jobs for each individual instance."""
 
+    @property
+    def num_machines(self) -> int:
+        """Returns the number of machines assigned to this multi-machine job."""
+        return len(self.machines)
+
     @property
     @abstractmethod
     def machine(self) -> "Machine":
@@ -303,7 +308,18 @@ class _BaseMMT(_BaseJob):
 
     def dict(
         self
-    ) -> Dict[str, Union[str, "Studio", "Status", "Machine", None, List[Dict[str, Union[str, "Status", "Machine"]]]]]:
+    ) -> Dict[
+        str,
+        Union[
+            str,
+            float,
+            "Studio",
+            "Status",
+            "Machine",
+            None,
+            List[Dict[str, Union[str, "Status", "Machine"]]],
+        ],
+    ]:
         """Dict representation of this job."""
         studio = self.studio
 
@@ -319,6 +335,7 @@ class _BaseMMT(_BaseJob):
                 {"name": d["name"], "status": d["status"], "machine": d["machine"]}
                 for d in (x.dict() for x in self.machines)
             ],
+            "total_cost": self.total_cost,
         }
 
     @abstractmethod

lightning_sdk/mmt/v1.py

@@ -4,11 +4,11 @@ from lightning_sdk.api.mmt_api import MMTApiV1
 from lightning_sdk.api.utils import _get_cloud_url
 from lightning_sdk.job.v1 import _internal_status_to_external_status
 from lightning_sdk.job.work import Work
+from lightning_sdk.status import Status
 
 if TYPE_CHECKING:
     from lightning_sdk.machine import Machine
     from lightning_sdk.organization import Organization
-    from lightning_sdk.status import Status
     from lightning_sdk.studio import Studio
     from lightning_sdk.teamspace import Teamspace
     from lightning_sdk.user import User
@@ -133,6 +133,8 @@ class _MMTV1(_BaseMMT):
 
     def stop(self) -> None:
         """Stops the job."""
+        if self.status in (Status.Stopped, Status.Completed, Status.Failed):
+            return
         self._job_api.stop_job(self._guaranteed_job.id, self.teamspace.id)
 
     def delete(self) -> None:

lightning_sdk/mmt/v2.py

@@ -1,13 +1,13 @@
-from typing import TYPE_CHECKING, Any, Dict, Optional, Tuple, Union
+from typing import TYPE_CHECKING, Dict, Optional, Tuple, Union
 
 from lightning_sdk.api.mmt_api import MMTApiV2
 from lightning_sdk.api.utils import _get_cloud_url
+from lightning_sdk.status import Status
 
 if TYPE_CHECKING:
     from lightning_sdk.job.job import Job
     from lightning_sdk.machine import Machine
     from lightning_sdk.organization import Organization
-    from lightning_sdk.status import Status
     from lightning_sdk.studio import Studio
     from lightning_sdk.teamspace import Teamspace
     from lightning_sdk.user import User
@@ -137,6 +137,8 @@ class _MMTV2(_BaseMMT):
 
     def stop(self) -> None:
         """Stops the job."""
+        if self.status in (Status.Stopped, Status.Completed, Status.Failed):
+            return
         self._job_api.stop_job(job_id=self._guaranteed_job.id, teamspace_id=self._teamspace.id)
 
     def delete(self) -> None:
@@ -149,12 +151,6 @@ class _MMTV2(_BaseMMT):
             teamspace_id=self._teamspace.id,
         )
 
-    @property
-    def _latest_job(self) -> Any:
-        """Guarantees to fetch the latest version of a job before returning it."""
-        self._update_internal_job()
-        return self._job
-
     @property
     def status(self) -> "Status":
         """The current status of the job."""
@@ -220,3 +216,8 @@ class _MMTV2(_BaseMMT):
     def command(self) -> str:
         """The command the job is running."""
         return self._job_api.get_command(self._guaranteed_job)
+
+    @property
+    def num_machines(self) -> int:
+        """Returns the number of machines assigned to this multi-machine job."""
+        return self._job_api.get_num_machines(self._guaranteed_job)

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

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

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

@@ -1,10 +1,10 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
-lightning_sdk/__init__.py,sha256=e-z42Ud9d3en8_MesDM8ZmP-uq7DC8lW5Y0dMWI9IUM,925
+lightning_sdk/__init__.py,sha256=pghaQ6_YR2SmOR4N7VM5mlRGqecB5VgWJpqV0JMLxiE,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=MLdhLtKjRICDikdXfFUMAW39kC-oC9IONGUv2yh8gRU,2090
+lightning_sdk/lit_container.py,sha256=Yh24NmVBsixKSYIZ_p0QqcZ0kjJO58OvXYDLC7ptx8w,2954
 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,39 +18,39 @@ 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=ykok7Fg3iNOKysgO601gt8gidXc9vO8IRqpjJel9lS8,11980
-lightning_sdk/api/lit_container_api.py,sha256=BF5AOvHmF8i2eYBIRgIKcS1R_y3wn2PL_LhZruQyEAo,821
-lightning_sdk/api/mmt_api.py,sha256=rgMVDdlC3h-KqGI2FGp7Im99gnrHg5xx4FLwQqYWqxI,7136
+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/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=pvDK_qUv8kgDTGumN2f7YcCy-HNnn_VUwZ5SbZTDgxA,22530
+lightning_sdk/api/utils.py,sha256=VG3sCxWjn6MHONYghA73PZH-ErgqsHyj1ZMU6qzBCFk,22762
 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/entrypoint.py,sha256=8CRA_jAg6CkHmFWUbrDQRRYd-oJccE6A7DPvNoAdGtk,1803
+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
 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/legacy.py,sha256=ocTVNwlsLRS5aMjbMkwFPjT3uEYvS8C40CJ0PeRRv8g,4707
-lightning_sdk/cli/list.py,sha256=sQzzwP6CxiL5YO4JZMgZ-THE4NII4h7mMl_9tmPZ_-8,2109
+lightning_sdk/cli/list.py,sha256=gLfIf9iXPPyotqIy8MkWGJXWmGF8BV3jtmcZ8cCd3Ng,3856
 lightning_sdk/cli/mmts_menu.py,sha256=OAwWywYeFSdmXBLR7raZlL7tpAhZld4r71LUHvmnZdA,2175
-lightning_sdk/cli/run.py,sha256=0jfijJCzUYON-OGgCljk6nhzN4o0pS2f8Il1LEOEPKk,11620
+lightning_sdk/cli/run.py,sha256=4hcPuawGPVoHMoLUw6uwTNv1qPd4Js-PomUJgjmLkfw,11806
 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
 lightning_sdk/cli/teamspace_menu.py,sha256=GVTrMAqxxL3INX83alo5wybnrl6rBBKENo01ZNkWPik,3753
-lightning_sdk/cli/upload.py,sha256=H9OyipYTYAQ9Mzy2e8jtoaa-B34-uXHbTQTzY2Vmhv4,9078
+lightning_sdk/cli/upload.py,sha256=pn7yIq98e_gMoaFhETHfiXai_15qdyG6t1hsdjN-oWg,10589
 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=CaysK2JiWTlTp7WJfC1Ebht2auCrMncgqn6EKVf-c_Q,15615
+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=pmu-adVC0fh4jKwXvjBQb9iIDKPuC5eOo0a2BiBq1fw,10184
-lightning_sdk/job/v2.py,sha256=5DhGK9KQhkRH-M3qzALhO83T1L0w9LEziaX4AFjL1Jo,10285
+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/lightning_cloud/__init__.py,sha256=o91SMAlwr4Ke5ESe8fHjqXcj31_h7rT-MlFoXA-n2EI,173
 lightning_sdk/lightning_cloud/__version__.py,sha256=lOfmWHtjmiuSG28TbKQqd2B3nwmSGOlKVFwhaj_cRJk,23
@@ -301,7 +301,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=PnRnqmq4FxzpreRvJ34Bc1jGp9k5-AQW-aLDl93HNqY,18931
+lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py,sha256=xGy5pWDcuKn1p90M4BGFlhAscswRMRmfRIRHdItR9aQ,17392
 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
@@ -528,7 +528,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_job_timing.py,sha256=qnnGfuKbAU_
 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
 lightning_sdk/lightning_cloud/openapi/models/v1_knowledge_configuration.py,sha256=zEm-VKsmvL6aJtICeq1w3nn-V-6GvYLRNYDbnvPGn-4,8841
-lightning_sdk/lightning_cloud/openapi/models/v1_lambda_labs_direct_v1.py,sha256=fYXcwvKIgeNWWCrtynvO5YCgbbPu0ETR6TtPnGbAPqY,4133
+lightning_sdk/lightning_cloud/openapi/models/v1_lambda_labs_direct_v1.py,sha256=aI89GSYPsCDMH2Xuk4bTcBmRDwv3ZQtvv7PqYT0TT3w,5290
 lightning_sdk/lightning_cloud/openapi/models/v1_lightning_app_user.py,sha256=5jJlea7GP24fkRtNsTVl0AYrotiODCYe8-qyIxr3w0w,4352
 lightning_sdk/lightning_cloud/openapi/models/v1_lightning_auth.py,sha256=-ccSh5ZorAALrYlIw6HhbFZ1NpF6YiU0WHuV1EnyBYM,4448
 lightning_sdk/lightning_cloud/openapi/models/v1_lightning_basic_auth.py,sha256=fgPODETba7KvV3JGrpgy44c6nXMt-rOHiK40SCgB0gI,4434
@@ -815,7 +815,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=2KlY1j_YxyEb228z8YLiDssXc1FFfwUFKk62X7T6RFE,73647
+lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py,sha256=BVJiD5oLzTBceVjdk2qS0_eA4xG4ANQzQAQNsoAYG-c,74282
 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
@@ -829,7 +829,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_validate_managed_endpoint_respon
 lightning_sdk/lightning_cloud/openapi/models/v1_validate_managed_model_response.py,sha256=LshfheXdKDqRYgg5Sa5dV0L9CW3BVfFts_f2qqmnH7k,3767
 lightning_sdk/lightning_cloud/openapi/models/v1_verify_verification_response.py,sha256=VQrBJk2pQQJuu-1Om-kvKr5N9ioxEPeV2l70udHEWSw,3058
 lightning_sdk/lightning_cloud/openapi/models/v1_volume.py,sha256=UIEsGm_TA6Qustavo2Umxq1No4pP-VImzsYw7CfNXMk,4702
-lightning_sdk/lightning_cloud/openapi/models/v1_vultr_direct_v1.py,sha256=jq4ODNqWXJqkkyMdod6C7eiuWDPAfqMC1sEI6DSJDH0,5594
+lightning_sdk/lightning_cloud/openapi/models/v1_vultr_direct_v1.py,sha256=5xp2xBJR823Xa-lTBf0bs9E1OnrhAidknF_LPnzgvuI,6465
 lightning_sdk/lightning_cloud/openapi/models/v1_work.py,sha256=MwXPSG8TE_CcGiffx6OI6U7kyf1NnSsDvJEM6LuwcYA,4910
 lightning_sdk/lightning_cloud/openapi/models/validate.py,sha256=wrCNhZpBR2PiA_a56C01w5fuxOSSGkbEh3Y8iMwUcYs,8086
 lightning_sdk/lightning_cloud/openapi/models/validateautojoindomain_domain_body.py,sha256=gRI7XdvbY7w09gIH6BkFai8UA-y71T2AmiR5Y2XcTe8,3781
@@ -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=yXmussF0qEh7S9KeYcKo7xzBlIAg1DCjgdf1iWc-L6M,14489
+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=3c28zivn2ZVhp4uXsyNwTpa4nN3vU021AyR6PCCw5_g,9255
-lightning_sdk/mmt/v2.py,sha256=LUwNSxeJVorCkJllRhFGtDgjGRBAsx81lLTaJWSFXHQ,9589
+lightning_sdk/mmt/v1.py,sha256=qdd8wrGlyg-75rYLGFY4-PfIg_O9fYvVRL-xryNg2U8,9347
+lightning_sdk/mmt/v2.py,sha256=97sIFx55MAgkaOq3pb5I8GB8pQoHg7HB8pezccH-vuw,9682
 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.48.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
-lightning_sdk-0.1.48.dist-info/METADATA,sha256=8eqGkORoyWrF3suijKWe2UnwH71uucciC5ajYlGemzk,4031
-lightning_sdk-0.1.48.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-lightning_sdk-0.1.48.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
-lightning_sdk-0.1.48.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
-lightning_sdk-0.1.48.dist-info/RECORD,,
+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,,