lightning-sdk 0.1.43__py3-none-any.whl → 0.1.44__py3-none-any.whl

lightning_sdk/__init__.py

@@ -27,5 +27,5 @@ __all__ = [
     "AIHub",
 ]
 
-__version__ = "0.1.43"
+__version__ = "0.1.44"
 _check_version_and_prompt_upgrade(__version__)

lightning_sdk/api/job_api.py

@@ -1,10 +1,12 @@
 import time
 from typing import TYPE_CHECKING, Dict, List, Optional
+from urllib.request import urlopen
 
 from lightning_sdk.api.utils import (
     _COMPUTE_NAME_TO_MACHINE,
     _MACHINE_TO_COMPUTE_NAME,
     _create_app,
+    remove_datetime_prefix,
 )
 from lightning_sdk.api.utils import (
     _get_cloud_url as _cloud_url,
@@ -16,7 +18,10 @@ from lightning_sdk.lightning_cloud.openapi import (
     Externalv1Lightningwork,
     JobsIdBody1,
     ProjectIdJobsBody,
+    V1CloudSpace,
     V1ComputeConfig,
+    V1DownloadJobLogsResponse,
+    V1DownloadLightningappInstanceLogsResponse,
     V1EnvVar,
     V1Job,
     V1JobSpec,
@@ -102,6 +107,12 @@ class JobApiV1:
         compute: str = compute_config.instance_type
         return _COMPUTE_NAME_TO_MACHINE[compute]
 
+    def get_studio_name(self, job: Externalv1LightningappInstance) -> str:
+        cs: V1CloudSpace = self._client.cloud_space_service_get_cloud_space(
+            project_id=job.project_id, id=job.spec.cloud_space_id
+        )
+        return cs.name
+
     def submit_job(
         self,
         name: str,
@@ -150,6 +161,16 @@ class JobApiV1:
 
         return Status.Pending
 
+    def get_logs_finished(self, job_id: str, work_id: str, teamspace_id: str) -> str:
+        resp: (
+            V1DownloadLightningappInstanceLogsResponse
+        ) = self._client.lightningapp_instance_service_download_lightningapp_instance_logs(
+            project_id=teamspace_id, id=job_id, work_id=work_id
+        )
+
+        data = urlopen(resp.url).read().decode("utf-8")
+        return remove_datetime_prefix(str(data))
+
 
 class JobApiV2:
     v2_job_state_pending = "pending"
@@ -247,6 +268,20 @@ class JobApiV2:
     def delete_job(self, job_id: str, teamspace_id: str, cloudspace_id: Optional[str]) -> None:
         self._client.jobs_service_delete_job(project_id=teamspace_id, id=job_id, cloudspace_id=cloudspace_id or "")
 
+    def get_logs_finished(self, job_id: str, teamspace_id: str) -> str:
+        resp: V1DownloadJobLogsResponse = self._client.jobs_service_download_job_logs(
+            project_id=teamspace_id, id=job_id
+        )
+
+        data = urlopen(resp.url).read().decode("utf-8")
+        return remove_datetime_prefix(str(data))
+
+    def get_studio_name(self, job: V1Job) -> str:
+        cs: V1CloudSpace = self._client.cloud_space_service_get_cloud_space(
+            project_id=job.project_id, id=job.spec.cloudspace_id
+        )
+        return cs.name
+
     def _job_state_to_external(self, state: str) -> "Status":
         from lightning_sdk.status import Status
 

lightning_sdk/api/utils.py

@@ -2,6 +2,7 @@ import concurrent.futures
 import errno
 import math
 import os
+import re
 from concurrent.futures import ThreadPoolExecutor
 from functools import partial
 from pathlib import Path
@@ -599,3 +600,10 @@ def _create_app(
         print(f"Create App: {resp.id=} {teamspace_id=} {studio_id=} {cloud_account=}")
 
     return resp
+
+
+def remove_datetime_prefix(text: str) -> str:
+    # Use a regular expression to match the datetime pattern at the start of each line
+    # lines looks something like
+    # '[2025-01-08T14:15:03.797142418Z] ⚡  ~ echo Hello\n[2025-01-08T14:15:03.803077717Z] Hello\n'
+    return re.sub(r"^\[.*?\] ", "", text, flags=re.MULTILINE)

lightning_sdk/job/base.py

@@ -273,6 +273,16 @@ class _BaseJob(ABC):
         """The teamspace the job is part of."""
         return self._teamspace
 
+    @property
+    @abstractmethod
+    def logs(self) -> str:
+        """The logs of the job."""
+
+    @property
+    def link(self) -> str:
+        """A link to view the current job in the UI."""
+        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=jobs&job_name={self.name}"
+
     @property
     def _guaranteed_job(self) -> Any:
         """Guarantees that the job was fetched at some point before returning it.

lightning_sdk/job/job.py

@@ -5,6 +5,9 @@ from lightning_sdk.api.user_api import UserApi
 from lightning_sdk.job.base import _BaseJob
 from lightning_sdk.job.v1 import _JobV1
 from lightning_sdk.job.v2 import _JobV2
+from lightning_sdk.utils.resolve import _setup_logger
+
+_logger = _setup_logger(__name__)
 
 if TYPE_CHECKING:
     from lightning_sdk.machine import Machine
@@ -27,6 +30,8 @@ def _has_jobs_v2() -> bool:
 class Job(_BaseJob):
     """Class to submit and manage single-machine jobs on the Lightning AI Platform."""
 
+    _force_v1: bool = False
+
     def __init__(
         self,
         name: str,
@@ -45,7 +50,7 @@ class Job(_BaseJob):
             user: the name of the user owning the :param`teamspace`
                 in case it is owned directly by a user instead of an org.
         """
-        internal_job_cls = _JobV2 if _has_jobs_v2() else _JobV1
+        internal_job_cls = _JobV2 if _has_jobs_v2() and not self._force_v1 else _JobV1
 
         self._internal_job = internal_job_cls(
             name=name,
@@ -127,6 +132,8 @@ class Job(_BaseJob):
         )
         # required for typing with "Job"
         assert isinstance(ret_val, cls)
+
+        _logger.info(f"Job was successfully launched. View it at {ret_val.link}")
         return ret_val
 
     def _submit(
@@ -239,9 +246,11 @@ class Job(_BaseJob):
         return self._internal_job._teamspace
 
     @property
-    def cluster(self) -> Optional[str]:
-        """The cluster the job is running on."""
-        return self._internal_job.cluster
+    def logs(self) -> str:
+        """The logs of the job."""
+        if self.status not in (Status.Failed, Status.Completed, Status.Stopped):
+            raise RuntimeError("Getting jobs logs while the job is pending or running is not supported yet!")
+        return self._internal_job.logs
 
     def __getattr__(self, key: str) -> Any:
         """Forward the attribute lookup to the internal job implementation."""
@@ -249,3 +258,7 @@ class Job(_BaseJob):
             return getattr(super(), key)
         except AttributeError:
             return getattr(self._internal_job, key)
+
+    @property
+    def link(self) -> str:
+        return self._internal_job.link

lightning_sdk/job/v1.py

@@ -210,6 +210,11 @@ class _JobV1(_BaseJob):
         """The path to the share of the job in the distributed teamspace filesystem."""
         return f"/teamspace/jobs/{self.name}/share"
 
+    @property
+    def logs(self) -> str:
+        """The logs of the job."""
+        return self.work.logs
+
     # the following and functions are solely to make the Work class function
     @property
     def _id(self) -> str:

lightning_sdk/job/v2.py

@@ -167,6 +167,24 @@ class _JobV2(_BaseJob):
         """The path to the share of the job within the distributed teamspace filesystem."""
         raise NotImplementedError("Not implemented yet")
 
+    @property
+    def logs(self) -> str:
+        from lightning_sdk.status import Status
+
+        if self.status not in (Status.Failed, Status.Completed, Status.Stopped):
+            raise RuntimeError("Getting jobs logs while the job is pending or running is not supported yet!")
+
+        return self._job_api.get_logs_finished(job_id=self._guaranteed_job.id, teamspace_id=self.teamspace.id)
+
+    @property
+    def link(self) -> str:
+        if self._guaranteed_job.spec.image:
+            return (
+                f"https://lightning.ai/{self.teamspace.owner.name}/{self.teamspace.name}/jobs/{self.name}?app_id=jobs"
+            )
+
+        return super().link
+
     def _update_internal_job(self) -> None:
         if getattr(self, "_job", None) is None:
             self._job = self._job_api.get_job_by_name(name=self._name, teamspace_id=self._teamspace.id)

lightning_sdk/job/work.py

@@ -60,3 +60,13 @@ class Work:
     @property
     def status(self) -> "Status":
         return self._job_api.get_status_from_work(self._latest_work)
+
+    @property
+    def logs(self) -> str:
+        """The logs of the work."""
+        from lightning_sdk.status import Status
+
+        if self.status not in (Status.Failed, Status.Completed, Status.Stopped):
+            raise RuntimeError("Getting jobs logs while the job is pending or running is not supported yet!")
+
+        return self._job_api.get_logs_finished(job_id=self._job._id, work_id=self._id, teamspace_id=self._teamspace.id)

lightning_sdk/lightning_cloud/openapi/models/v1_membership.py

@@ -52,6 +52,7 @@ class V1Membership(object):
         'job_count': 'str',
         'membership_count': 'str',
         'name': 'str',
+        'next_free_credits_grant_at': 'datetime',
         'owner_id': 'str',
         'owner_type': 'V1OwnerType',
         'project_id': 'str',
@@ -73,6 +74,7 @@ class V1Membership(object):
         'job_count': 'jobCount',
         'membership_count': 'membershipCount',
         'name': 'name',
+        'next_free_credits_grant_at': 'nextFreeCreditsGrantAt',
         'owner_id': 'ownerId',
         'owner_type': 'ownerType',
         'project_id': 'projectId',
@@ -82,7 +84,7 @@ class V1Membership(object):
         'user_id': 'userId'
     }
 
-    def __init__(self, balance: 'float' =None, created_at: 'datetime' =None, creator_id: 'str' =None, datastore_count: 'str' =None, description: 'str' =None, display_name: 'str' =None, free_credits_enabled: 'bool' =None, is_default: 'bool' =None, job_count: 'str' =None, membership_count: 'str' =None, name: 'str' =None, owner_id: 'str' =None, owner_type: 'V1OwnerType' =None, project_id: 'str' =None, quotas: 'V1Quotas' =None, roles: 'list[V1Role]' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, balance: 'float' =None, created_at: 'datetime' =None, creator_id: 'str' =None, datastore_count: 'str' =None, description: 'str' =None, display_name: 'str' =None, free_credits_enabled: 'bool' =None, is_default: 'bool' =None, job_count: 'str' =None, membership_count: 'str' =None, name: 'str' =None, next_free_credits_grant_at: 'datetime' =None, owner_id: 'str' =None, owner_type: 'V1OwnerType' =None, project_id: 'str' =None, quotas: 'V1Quotas' =None, roles: 'list[V1Role]' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
         """V1Membership - a model defined in Swagger"""  # noqa: E501
         self._balance = None
         self._created_at = None
@@ -95,6 +97,7 @@ class V1Membership(object):
         self._job_count = None
         self._membership_count = None
         self._name = None
+        self._next_free_credits_grant_at = None
         self._owner_id = None
         self._owner_type = None
         self._project_id = None
@@ -125,6 +128,8 @@ class V1Membership(object):
             self.membership_count = membership_count
         if name is not None:
             self.name = name
+        if next_free_credits_grant_at is not None:
+            self.next_free_credits_grant_at = next_free_credits_grant_at
         if owner_id is not None:
             self.owner_id = owner_id
         if owner_type is not None:
@@ -371,6 +376,27 @@ class V1Membership(object):
 
         self._name = name
 
+    @property
+    def next_free_credits_grant_at(self) -> 'datetime':
+        """Gets the next_free_credits_grant_at of this V1Membership.  # noqa: E501
+
+
+        :return: The next_free_credits_grant_at of this V1Membership.  # noqa: E501
+        :rtype: datetime
+        """
+        return self._next_free_credits_grant_at
+
+    @next_free_credits_grant_at.setter
+    def next_free_credits_grant_at(self, next_free_credits_grant_at: 'datetime'):
+        """Sets the next_free_credits_grant_at of this V1Membership.
+
+
+        :param next_free_credits_grant_at: The next_free_credits_grant_at of this V1Membership.  # noqa: E501
+        :type: datetime
+        """
+
+        self._next_free_credits_grant_at = next_free_credits_grant_at
+
     @property
     def owner_id(self) -> 'str':
         """Gets the owner_id of this V1Membership.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_state.py

@@ -38,6 +38,7 @@ class V1MultiMachineJobState(object):
     allowed enum values
     """
     UNSPECIFIED = "MultiMachineJob_STATE_UNSPECIFIED"
+    PENDING = "MultiMachineJob_STATE_PENDING"
     RUNNING = "MultiMachineJob_STATE_RUNNING"
     STOPPED = "MultiMachineJob_STATE_STOPPED"
     DELETED = "MultiMachineJob_STATE_DELETED"

lightning_sdk/lightning_cloud/openapi/models/v1_project_membership.py

@@ -58,6 +58,7 @@ class V1ProjectMembership(object):
         'last_name': 'str',
         'membership_count': 'str',
         'name': 'str',
+        'next_free_credits_grant_at': 'datetime',
         'organization': 'str',
         'owner_id': 'str',
         'owner_type': 'V1OwnerType',
@@ -87,6 +88,7 @@ class V1ProjectMembership(object):
         'last_name': 'lastName',
         'membership_count': 'membershipCount',
         'name': 'name',
+        'next_free_credits_grant_at': 'nextFreeCreditsGrantAt',
         'organization': 'organization',
         'owner_id': 'ownerId',
         'owner_type': 'ownerType',
@@ -98,7 +100,7 @@ class V1ProjectMembership(object):
         'username': 'username'
     }
 
-    def __init__(self, avatar_url: 'str' =None, balance: 'float' =None, created_at: 'datetime' =None, creator_id: 'str' =None, datastore_count: 'str' =None, description: 'str' =None, display_name: 'str' =None, email: 'str' =None, first_name: 'str' =None, free_credits_enabled: 'bool' =None, inactive: 'bool' =None, is_default: 'bool' =None, job_count: 'str' =None, job_title: 'str' =None, last_name: 'str' =None, membership_count: 'str' =None, name: 'str' =None, organization: 'str' =None, owner_id: 'str' =None, owner_type: 'V1OwnerType' =None, project_id: 'str' =None, quotas: 'V1Quotas' =None, roles: 'list[V1Role]' =None, updated_at: 'datetime' =None, user_id: 'str' =None, username: 'str' =None):  # noqa: E501
+    def __init__(self, avatar_url: 'str' =None, balance: 'float' =None, created_at: 'datetime' =None, creator_id: 'str' =None, datastore_count: 'str' =None, description: 'str' =None, display_name: 'str' =None, email: 'str' =None, first_name: 'str' =None, free_credits_enabled: 'bool' =None, inactive: 'bool' =None, is_default: 'bool' =None, job_count: 'str' =None, job_title: 'str' =None, last_name: 'str' =None, membership_count: 'str' =None, name: 'str' =None, next_free_credits_grant_at: 'datetime' =None, organization: 'str' =None, owner_id: 'str' =None, owner_type: 'V1OwnerType' =None, project_id: 'str' =None, quotas: 'V1Quotas' =None, roles: 'list[V1Role]' =None, updated_at: 'datetime' =None, user_id: 'str' =None, username: 'str' =None):  # noqa: E501
         """V1ProjectMembership - a model defined in Swagger"""  # noqa: E501
         self._avatar_url = None
         self._balance = None
@@ -117,6 +119,7 @@ class V1ProjectMembership(object):
         self._last_name = None
         self._membership_count = None
         self._name = None
+        self._next_free_credits_grant_at = None
         self._organization = None
         self._owner_id = None
         self._owner_type = None
@@ -161,6 +164,8 @@ class V1ProjectMembership(object):
             self.membership_count = membership_count
         if name is not None:
             self.name = name
+        if next_free_credits_grant_at is not None:
+            self.next_free_credits_grant_at = next_free_credits_grant_at
         if organization is not None:
             self.organization = organization
         if owner_id is not None:
@@ -537,6 +542,27 @@ class V1ProjectMembership(object):
 
         self._name = name
 
+    @property
+    def next_free_credits_grant_at(self) -> 'datetime':
+        """Gets the next_free_credits_grant_at of this V1ProjectMembership.  # noqa: E501
+
+
+        :return: The next_free_credits_grant_at of this V1ProjectMembership.  # noqa: E501
+        :rtype: datetime
+        """
+        return self._next_free_credits_grant_at
+
+    @next_free_credits_grant_at.setter
+    def next_free_credits_grant_at(self, next_free_credits_grant_at: 'datetime'):
+        """Sets the next_free_credits_grant_at of this V1ProjectMembership.
+
+
+        :param next_free_credits_grant_at: The next_free_credits_grant_at of this V1ProjectMembership.  # noqa: E501
+        :type: datetime
+        """
+
+        self._next_free_credits_grant_at = next_free_credits_grant_at
+
     @property
     def organization(self) -> 'str':
         """Gets the organization of this V1ProjectMembership.  # noqa: E501

lightning_sdk/mmt/base.py

@@ -36,6 +36,11 @@ class MMTMachine(Protocol):
         """The status of this job."""
         ...
 
+    @property
+    def logs(self) -> str:
+        """The logs of the given machine."""
+        ...
+
 
 class _BaseMMT(_BaseJob):
     """Base interface to all job types."""
@@ -273,6 +278,11 @@ class _BaseMMT(_BaseJob):
         """The teamspace the job is part of."""
         return self._teamspace
 
+    @property
+    def logs(self) -> str:
+        """Logs of the rank 0 machine."""
+        return self.machines[0].logs
+
     @abstractmethod
     def _update_internal_job(self) -> None:
         pass

lightning_sdk/mmt/mmt.py

@@ -1,3 +1,4 @@
+from contextlib import suppress
 from functools import lru_cache
 from typing import TYPE_CHECKING, Any, Dict, Optional, Tuple, Union
 
@@ -6,6 +7,7 @@ from lightning_sdk.job.job import _has_jobs_v2
 from lightning_sdk.mmt.base import MMTMachine, _BaseMMT
 from lightning_sdk.mmt.v1 import _MMTV1
 from lightning_sdk.mmt.v2 import _MMTV2
+from lightning_sdk.utils.resolve import _setup_logger
 
 if TYPE_CHECKING:
     from lightning_sdk.machine import Machine
@@ -15,6 +17,8 @@ if TYPE_CHECKING:
     from lightning_sdk.teamspace import Teamspace
     from lightning_sdk.user import User
 
+_logger = _setup_logger(__name__)
+
 
 @lru_cache(maxsize=None)
 def _has_mmt_v2() -> bool:
@@ -137,8 +141,15 @@ class MMT(_BaseMMT):
             artifacts_remote=artifacts_remote,
             cluster=cluster,  # deprecated in favor of cloud_account
         )
-        # required for typing with "Job"
+        # required for typing with "MMT"
         assert isinstance(ret_val, cls)
+
+        msg = "Multi-Machine Job was successfully launched."
+
+        with suppress(NotImplementedError):
+            msg += f" View it at {ret_val.link}"
+
+        _logger.info(msg)
         return ret_val
 
     def _submit(
@@ -255,6 +266,10 @@ class MMT(_BaseMMT):
         """The teamspace the job is part of."""
         return self._internal_mmt._teamspace
 
+    @property
+    def link(self) -> str:
+        return self._internal_mmt.link
+
     def __getattr__(self, key: str) -> Any:
         """Forward the attribute lookup to the internal job implementation."""
         try:

lightning_sdk/mmt/v1.py

@@ -172,6 +172,10 @@ class _MMTV1(_BaseMMT):
         """The teamspace the job is part of."""
         return self._teamspace
 
+    @property
+    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}"
+
     # the following and functions are solely to make the Work class function
     @property
     def _id(self) -> str:

lightning_sdk/mmt/v2.py

@@ -186,3 +186,8 @@ class _MMTV2(_BaseMMT):
     def teamspace(self) -> "Teamspace":
         """The teamspace the job is part of."""
         return self._teamspace
+
+    @property
+    def link(self) -> str:
+        # TODO: Since we don't have a UI for this yet, we can't have a link
+        raise NotImplementedError

lightning_sdk/plugin.py

@@ -3,7 +3,8 @@ import logging
 import os
 import warnings
 from abc import ABC, abstractmethod
-from typing import TYPE_CHECKING, Any, Optional, Protocol, Union, runtime_checkable
+from contextlib import contextmanager
+from typing import TYPE_CHECKING, Any, Generator, Optional, Protocol, Union, runtime_checkable
 
 from lightning_sdk.job import Job
 from lightning_sdk.machine import Machine
@@ -16,6 +17,7 @@ from lightning_sdk.utils.resolve import (
 
 if TYPE_CHECKING:
     from lightning_sdk.lightning_cloud.openapi import Externalv1LightningappInstance
+    from lightning_sdk.mmt import MMT
 
 _logger = _setup_logger(__name__)
 
@@ -141,7 +143,7 @@ class JobsPlugin(_Plugin):
 
         machine = _resolve_deprecated_cloud_compute(machine, cloud_compute)
 
-        job = Job.run(
+        return Job.run(
             name=name,
             machine=machine,
             command=command,
@@ -151,10 +153,6 @@ class JobsPlugin(_Plugin):
             interruptible=interruptible,
         )
 
-        _logger.info(_success_message(job, self))
-
-        return job
-
 
 class MultiMachineTrainingPlugin(_Plugin):
     """Plugin handling multi-machine-training jobs."""
@@ -170,7 +168,7 @@ class MultiMachineTrainingPlugin(_Plugin):
         cloud_compute: Optional[Machine] = None,
         num_instances: int = 2,
         interruptible: bool = False,
-    ) -> Job:
+    ) -> "MMT":
         """Launches an asynchronous multi-machine-training.
 
         Args:
@@ -188,20 +186,16 @@ class MultiMachineTrainingPlugin(_Plugin):
 
         machine = _resolve_deprecated_cloud_compute(machine, cloud_compute)
 
-        MMT._force_v1 = True
-
-        mmt = MMT.run(
-            name=name,
-            num_machines=num_instances,
-            machine=machine,
-            command=command,
-            studio=self._studio,
-            teamspace=self._studio.teamspace,
-            interruptible=interruptible,
-        )
-
-        MMT._force_v1 = False
-        return mmt
+        with forced_v1(MMT) as v1mmt:
+            return v1mmt.run(
+                name=name,
+                num_machines=num_instances,
+                machine=machine,
+                command=command,
+                studio=self._studio,
+                teamspace=self._studio.teamspace,
+                interruptible=interruptible,
+            )
 
 
 class MultiMachineDataPrepPlugin(_Plugin):
@@ -245,7 +239,8 @@ class MultiMachineDataPrepPlugin(_Plugin):
             interruptible=interruptible,
         )
 
-        return Job(resp.name, self._studio.teamspace)
+        with forced_v1(Job) as v1_job:
+            return v1_job(resp.name, self._studio.teamspace)
 
 
 class InferenceServerPlugin(_Plugin):
@@ -293,7 +288,8 @@ class InferenceServerPlugin(_Plugin):
         )
 
         _logger.info(_success_message(resp, self))
-        return Job(resp.name, self._studio.teamspace)
+        with forced_v1(Job) as v1_job:
+            return v1_job(resp.name, self._studio.teamspace)
 
 
 class SlurmJobsPlugin(_Plugin):
@@ -430,3 +426,12 @@ def _run_name(plugin_type: str) -> str:
 def _success_message(resp: Union["Externalv1LightningappInstance", Job], plugin_instance: _RunnablePlugin) -> str:
     """Compiles the success message for a given runnable plugin."""
     return f"{plugin_instance._plugin_run_name} {resp.name} was successfully launched. View it at https://lightning.ai/{plugin_instance._studio.owner.name}/{plugin_instance._studio.teamspace.name}/studios/{plugin_instance.studio}/app?app_id={plugin_instance._slug_name}&job_name={resp.name}"
+
+
+@contextmanager
+def forced_v1(cls: Any) -> Generator[Any, None, None]:
+    """Forces to use the v1 version of a class when using a class with multiple backends."""
+    orig_val = getattr(cls, "_force_v1", False)
+    cls._force_v1 = True
+    yield cls
+    cls._force_v1 = orig_val

{lightning_sdk-0.1.43.dist-info → lightning_sdk-0.1.44.dist-info}/METADATA

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

{lightning_sdk-0.1.43.dist-info → lightning_sdk-0.1.44.dist-info}/RECORD

@@ -1,5 +1,5 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
-lightning_sdk/__init__.py,sha256=je3-DFVBfoAqqPLrFLZmVZGWPNda7886O9-BrY6dvlQ,925
+lightning_sdk/__init__.py,sha256=XJlg9TRR7l1VeCSLkPRsB0dOcFAU6kcMciUwmYdtojQ,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
@@ -8,7 +8,7 @@ lightning_sdk/machine.py,sha256=VdFXStR6ilYBEYuxgGWzcAw2TtW-nEQVsh6hz-2aaEw,750
 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=cDvXaQWyove0IdEZYkvlKiaKuQoyjmcUJ8WdZkZ61Z0,14512
+lightning_sdk/plugin.py,sha256=nWFL1l6Q9DE8Lr2krD5qRhZljwmYm00R2eR1tYRb20s,14902
 lightning_sdk/status.py,sha256=kLDhN4-zdsGuZM577JMl1BbUIoF61bUOadW89ZAATFA,219
 lightning_sdk/studio.py,sha256=lezGs111RUFejLWgp4Urov5l6uiUmYJjtRK8D8EYFU8,17198
 lightning_sdk/teamspace.py,sha256=dKT-WrYF2xGP1C1bjOY2aYlEpkrvYkz2fXvtYVgogwo,11508
@@ -17,13 +17,13 @@ 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=I90j2DAhokqbHoyORKI0-dgrsZLvUaFHa8pT49IM60c,9683
+lightning_sdk/api/job_api.py,sha256=KgXl0uO32Ja0AvxOASh-LihUPGERq2Fwy1rovXnq5Sg,11074
 lightning_sdk/api/mmt_api.py,sha256=texQJqSjbQNpfLLrumpvZ0MauPjmBlJAc8oSk3i46wk,6569
 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/user_api.py,sha256=sL7RIjjtmZmvCZWx7BBZslhj1BeNh4Idn-RVcdmf7M0,2598
-lightning_sdk/api/utils.py,sha256=iGMi5mabu4NyY9EPZv6HpbbgR5qzLUq8XSyX2zWy7l8,22193
+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
@@ -37,11 +37,11 @@ lightning_sdk/cli/upload.py,sha256=H9OyipYTYAQ9Mzy2e8jtoaa-B34-uXHbTQTzY2Vmhv4,9
 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=I4-iWyiKp1KUkxDy97zJYbwbdQ_7cu6FqCayKwXDloQ,13000
-lightning_sdk/job/job.py,sha256=pT9rkSsK5BHu6dSyHpAlYfGXvaF0s_XtrjjhOSAVFWU,11070
-lightning_sdk/job/v1.py,sha256=zIcngaM2_c-2thcKiCOPoWGUuIKc4tasJEwEFJv6bAA,8953
-lightning_sdk/job/v2.py,sha256=oq54VFInuVV_L-nUO_dnBbn4TxPWiBuIqmdFpNN1LmU,8057
-lightning_sdk/job/work.py,sha256=PYopS_6c556I2o8ouSXmzb4FGQflzCe06GpqJiCdedw,1604
+lightning_sdk/job/base.py,sha256=WHd2jz1qEeQWg2ljphlJMwoEVU6qtSBjnO9hMScLF7E,13383
+lightning_sdk/job/job.py,sha256=8Bch1TCKjf4jxf46tVN64_k_ZIq677kblHLJHQJO8v4,11536
+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/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
@@ -632,7 +632,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_response.py,sha
 lightning_sdk/lightning_cloud/openapi/models/v1_managed_endpoint.py,sha256=T0IXmXRI5paAN6NxpQQ8873lgTT7bIHYZZmeJEmW4yA,10175
 lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py,sha256=97KxjHxTsaLWYlTNQeE2yGhF7KbTbmqLb8c39BEy02s,9117
 lightning_sdk/lightning_cloud/openapi/models/v1_managed_model_abilities.py,sha256=1of-AEIimGnUd_0fpS5U8apGFYdsIXogRpFGxed67s0,5892
-lightning_sdk/lightning_cloud/openapi/models/v1_membership.py,sha256=avU-MIUu2qy7tyjfVWmuPfrGOvZXK9GuirIKR1eu-CY,16140
+lightning_sdk/lightning_cloud/openapi/models/v1_membership.py,sha256=6wqHKmxv3SOj9mTTRAw2g3ljxQyg_HKU18rexmA1KII,17207
 lightning_sdk/lightning_cloud/openapi/models/v1_message.py,sha256=6E1FDyfNM8fMkJi6FZ4U9UBk4k5VhfmCYOHGkc5CyzA,10189
 lightning_sdk/lightning_cloud/openapi/models/v1_message_author.py,sha256=QRC1HofufQOPROrxwtLeaEfE8f1h_ehZ0tdqlYfkgyM,3591
 lightning_sdk/lightning_cloud/openapi/models/v1_message_content.py,sha256=rlYtO8Llal6jISWpQMyOC8zMOvdVWaljXTwlHp_gzdc,4535
@@ -652,7 +652,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_event.py,sha25
 lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_event_type.py,sha256=HMBQuVSxwwbhbDQAAAMF8k99drvllwifjwjqEQrrDaQ,3248
 lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_fault_tolerance.py,sha256=5Q0mZ3FM6yRi81mVCnYcyYh84loK4LVZTIv-2GgAy1I,4863
 lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_fault_tolerance_strategy.py,sha256=FQ8qqDLfpguNtMqEETybXre4IArM32_vFuLrrUOcHb4,3539
-lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_state.py,sha256=0Rh4GKRQcdfI0ccGeVt0PxTbC7e7QwaIWYUy1rCnVhE,3349
+lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_state.py,sha256=xNpq6pM1UF9r0hPHoEQ124SlCW2sCb_zW628kEa7IVU,3395
 lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_status.py,sha256=yoOMJoszmdfEoUP7mzjl3PTK9dq6JUi8yXrxJljrY-0,9540
 lightning_sdk/lightning_cloud/openapi/models/v1_named_get_logger_metrics.py,sha256=VygKKjSPGIbWT9RxbmJfOSlNR-Fq12yKj8wT_SHd2ZQ,3966
 lightning_sdk/lightning_cloud/openapi/models/v1_network_config.py,sha256=bHDZVIsqgiSi7eBf72RunvrLR-d-4b9S7oDMAZFBvCM,5476
@@ -682,7 +682,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_project_artifact.py,sha256=jMg7B
 lightning_sdk/lightning_cloud/openapi/models/v1_project_cluster_binding.py,sha256=KLZYf_bkkB1OJenDpZsG02BvomPQqJSP3gMf3ofJ83k,8719
 lightning_sdk/lightning_cloud/openapi/models/v1_project_compute_daily_usage.py,sha256=92qqkHCx9gfZCg7FGVn7TDMQjuHw5E0POtLC7HoQ9pE,6565
 lightning_sdk/lightning_cloud/openapi/models/v1_project_compute_usage.py,sha256=-IJmDR0j5M2S7wfkwz5qO0riW2ZolAzrLnopnBAfckU,12121
-lightning_sdk/lightning_cloud/openapi/models/v1_project_membership.py,sha256=_9GewIzuL3F3Nn9duxOAq1mI7f2AfODWgVhvVNqfCsI,22500
+lightning_sdk/lightning_cloud/openapi/models/v1_project_membership.py,sha256=pzF0Jm8gNDV1-TerXtPWf1xNM2pRgJnCtDtKNRWcb8o,23595
 lightning_sdk/lightning_cloud/openapi/models/v1_project_membership_invite.py,sha256=V9Djs6edeaKv-X2-GpfAU-EVnx2rlKppGSTLfKzfrP0,9310
 lightning_sdk/lightning_cloud/openapi/models/v1_project_membership_role_binding.py,sha256=pS5454riAOUNeB7B3Y8ChEj9RJQYaroelZuJQs8w2yM,7727
 lightning_sdk/lightning_cloud/openapi/models/v1_project_settings.py,sha256=OugDsaWQPBMBGx4MykBJZO-xacRzzVMV3uGbeyLtQi0,5977
@@ -834,10 +834,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=lUB8pAKXTjn_WD6vcJIUMxtZQwrzwxYnaPJbaiPPtMw,12548
-lightning_sdk/mmt/mmt.py,sha256=vMAoR3qjb0VRwCXpKpAMxeTEMALmQB9_RK064fTHM8M,11922
-lightning_sdk/mmt/v1.py,sha256=8LjZnMSGgsGLeajuPto3gknJwVRvsGfkYVoo5A_UlO8,7917
-lightning_sdk/mmt/v2.py,sha256=TsJ8PNluyU0WnNdHvEKWkpJxMwjmyFFJZIaJKCgugIM,8201
+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/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
@@ -846,9 +846,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.43.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
-lightning_sdk-0.1.43.dist-info/METADATA,sha256=qsepNbeCeaApc3hDy1vhjhXs-sIoKUwYfMVK_MaKVUA,4031
-lightning_sdk-0.1.43.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-lightning_sdk-0.1.43.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
-lightning_sdk-0.1.43.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
-lightning_sdk-0.1.43.dist-info/RECORD,,
+lightning_sdk-0.1.44.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-0.1.44.dist-info/METADATA,sha256=FpuwHfzu6vOmo_c85UKIO8Ky54Hm8l0gyUOgV84XC2Y,4031
+lightning_sdk-0.1.44.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+lightning_sdk-0.1.44.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
+lightning_sdk-0.1.44.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-0.1.44.dist-info/RECORD,,