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

lightning_sdk/utils/enum.py

@@ -0,0 +1,116 @@
+from enum import Enum, EnumMeta, unique
+from types import MappingProxyType
+from typing import Any, Dict, Iterator, Union
+
+
+# We silence annotation errors as the function signatures are exactly the same in python core enum.py
+# modified from https://stackoverflow.com/a/62309159
+class OnAccess(EnumMeta):
+    """Runs a user-specified function whenever member is accessed."""
+
+    def __getattribute__(cls, name: str) -> Any:  # noqa: N805
+        """Called when an attribute is accessed."""
+        obj = super().__getattribute__(name)
+        if isinstance(obj, Enum) and obj._on_access:
+            obj._on_access()
+        return _resolve_deprecated_enum_fields(obj)
+
+    def __getitem__(cls, name: str) -> Any:  # noqa: N805
+        """Called when an item is accessed."""
+        member = super().__getitem__(name)
+        if isinstance(member, Enum) and member._on_access:
+            member._on_access()
+
+        return _resolve_deprecated_enum_fields(member)
+
+    def __call__(
+        cls,  # noqa: N805
+        value,  # noqa: ANN001
+        names=None,  # noqa: ANN001
+        *,
+        module=None,  # noqa: ANN001
+        qualname=None,  # noqa: ANN001
+        type=None,  # noqa: A002, ANN001
+        start=1,  # noqa: ANN001
+    ) -> Any:
+        obj = super().__call__(value, names, module=module, qualname=qualname, type=type, start=start)
+        if isinstance(obj, Enum) and obj._on_access:
+            obj._on_access()
+        return _resolve_deprecated_enum_fields(obj)
+
+    @staticmethod
+    def _filter_func(member: "DeprecationEnum") -> bool:
+        return not member.deprecated
+
+    def __iter__(cls) -> Iterator["DeprecationEnum"]:  # noqa: N805
+        """Returns members in definition order.
+
+        Excludes deprecated members
+
+        """
+        return filter(cls._filter_func, (cls._member_map_[name] for name in cls._member_names_))
+
+    def __len__(cls) -> int:  # noqa: N805
+        """Length of the enum excluding deprecated members."""
+        return len(cls.__members__)
+
+    @property
+    def __members__(cls) -> Dict[str, "DeprecationEnum"]:  # noqa: N805
+        """Returns a mapping of member name->value.
+
+        This mapping lists all enum members, including aliases. Note that this is a read-only view of the internal
+        mapping.
+
+        Excludes deprecated members.
+
+        """
+        return MappingProxyType(
+            {name: member for name, member in cls._member_map_.items() if not cls._filter_func(member)}
+        )
+
+    def __reversed__(cls) -> Iterator["DeprecationEnum"]:  # noqa: N805
+        """Returns members in reverse definition order.
+
+        Excludes deprecated members
+
+        """
+        return filter(cls._filter_func, (cls._member_map_[name] for name in reversed(cls._member_names_)))
+
+
+def _resolve_deprecated_enum_fields(value: Union[Any, "DeprecationEnum"]) -> Union[Any, "DeprecationEnum"]:
+    """Resolves deprecated enum fields to their preferred values."""
+    if getattr(value, "deprecated", False):
+        return value.__class__[value.preferred_way]
+    return value
+
+
+# NOTE: even deprecated enums need to have unique values.
+# otherwise the deprecation warning will not be triggered
+# as duplicated values are handled as aliases towards the original value
+# and the original value is not accessed. However, _resolve_deprecated_enum_fields
+# will handle this case and return the preferred value
+@unique
+class DeprecationEnum(Enum, metaclass=OnAccess):
+    def __new__(cls, value, *args) -> Any:  # noqa: ANN001, ANN002
+        member = object.__new__(cls)
+        member._value_ = value
+        member.deprecated = False
+        member._on_access = None
+
+        if args:
+            member.preferred_way = args[0]
+            member._on_access = member.deprecate
+            member.deprecated = True
+
+        return member
+
+    def deprecate(self) -> None:
+        import warnings
+
+        clsname = self.__class__.__name__
+
+        warnings.warn(
+            f"{clsname}.{self.name} is deprecated. Use {clsname}.{self.preferred_way} instead!",
+            DeprecationWarning,
+            stacklevel=3,
+        )

lightning_sdk/{utils.py → utils/resolve.py}

@@ -2,7 +2,7 @@ import logging
 import os
 import warnings
 from contextlib import contextmanager
-from typing import TYPE_CHECKING, Generator, List, Optional, Union
+from typing import TYPE_CHECKING, Generator, List, Optional, Tuple, Union
 
 from lightning_sdk.api import TeamspaceApi, UserApi
 from lightning_sdk.machine import Machine
@@ -13,6 +13,9 @@ if TYPE_CHECKING:
     from lightning_sdk.user import User
 
 
+_LIGHTNING_SERVICE_EXECUTION_ID_KEY = "LIGHTNING_SERVICE_EXECUTION_ID"
+
+
 def _setup_logger(name: str, level: int = logging.INFO) -> logging.Logger:
     _logger = logging.getLogger(name)
     _handler = logging.StreamHandler()
@@ -43,6 +46,24 @@ def _resolve_deprecated_cloud_compute(machine: Machine, cloud_compute: Optional[
     return machine
 
 
+def _resolve_deprecated_cluster(cloud_account: Optional[str], cluster: Optional[str]) -> Optional[str]:
+    if cluster is not None:
+        if cloud_account is not None:
+            raise ValueError(
+                "Cannot use both 'cluster' and 'cloud_account' at the same time."
+                "Please don't set the 'cluster' as it will be deprecated!"
+            )
+
+        warnings.warn(
+            "The 'cluster' argument will be deprecated in the future! "
+            "Please consider using the 'cloud_account' argument instead!",
+            DeprecationWarning,
+        )
+        return cluster
+
+    return cloud_account
+
+
 def _resolve_org_name(name: Optional[str]) -> Optional[str]:
     if name is None:
         name = os.environ.get("LIGHTNING_ORG", "") or None
@@ -54,7 +75,6 @@ def _resolve_org(org: Optional[Union[str, "Organization"]]) -> Optional["Organiz
 
     if isinstance(org, Organization):
         return org
-
     org = _resolve_org_name(org)
 
     if org is None:
@@ -68,7 +88,6 @@ def _resolve_org(org: Optional[Union[str, "Organization"]]) -> Optional["Organiz
 def _resolve_user_name(name: Optional[str]) -> Optional[str]:
     if name is None:
         name = os.environ.get("LIGHTNING_USERNAME", "") or None
-
     return name
 
 
@@ -79,7 +98,6 @@ def _resolve_user(user: Optional[Union[str, "User"]]) -> Optional["User"]:
         return user
 
     user = _resolve_user_name(user)
-
     if user is None:
         return None
 
@@ -131,6 +149,12 @@ def _get_organizations_for_authed_user() -> List["Organization"]:
     return [Organization(_org.name) for _org in _orgs]
 
 
+def _get_teamspace_names_for_authed_user() -> List[str]:
+    """Returns Teamspace's names the current Authed user is a member of."""
+    teamspaces = UserApi()._get_all_teamspace_memberships("")
+    return sorted([ts.name for ts in teamspaces])
+
+
 def _get_authed_user() -> "User":
     from lightning_sdk.user import User
 
@@ -150,3 +174,16 @@ def skip_studio_init() -> Generator[None, None, None]:
     yield
 
     Studio._skip_init = prev_studio_init_state
+
+
+def _parse_model_and_version(name: str) -> Tuple[str, str]:
+    parts = name.split(":")
+    if len(parts) == 1:
+        return parts[0], "latest"
+    if len(parts) == 2:
+        return parts[0], parts[1]
+    # The rest of the validation for name and version happens in the backend
+    raise ValueError(
+        "Model version is expected to be in the format `entity/modelname:version` separated by a"
+        f" single colon, but got: {name}"
+    )

lightning_sdk-0.1.47.dist-info/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Grid.ai
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

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

@@ -1,14 +1,37 @@
 Metadata-Version: 2.1
 Name: lightning_sdk
-Version: 0.1.3
+Version: 0.1.47
 Summary: SDK to develop using Lightning AI Studios
 Author-email: Lightning-AI <justus@lightning.ai>
+License: MIT License
+        
+        Copyright (c) 2024 Grid.ai
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy
+        of this software and associated documentation files (the "Software"), to deal
+        in the Software without restriction, including without limitation the rights
+        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+        copies of the Software, and to permit persons to whom the Software is
+        furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all
+        copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+        SOFTWARE.
+        
 Keywords: deep learning,machine learning,pytorch,AI
 Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
+License-File: LICENSE
 Requires-Dist: backoff
-Requires-Dist: click
 Requires-Dist: fastapi
 Requires-Dist: pyjwt
 Requires-Dist: python-multipart
@@ -22,6 +45,9 @@ Requires-Dist: tqdm
 Requires-Dist: fire
 Requires-Dist: simple-term-menu
 Requires-Dist: lightning-utilities
+Provides-Extra: serve
+Requires-Dist: litserve>=0.2.5; extra == "serve"
+Requires-Dist: docker; extra == "serve"
 
 # Lightning SDK
 
@@ -70,8 +96,8 @@ s.install_plugin("jobs")
 s.install_plugin("multi-machine-training")
 
 # run the resulting plugins to start 1 job and 1 multi-machine training
-s.installed_plugins["jobs"].run("python my_dummy_file", name="my_first_job", cloud_compute=Machine.A10G)
-s.installed_plugins["multi-machine-training"].run("python my_dummy_file", name="my_first_mmt", cloud_compute=Machine.T4, num_instances=42)
+s.installed_plugins["jobs"].run("python my_dummy_file", name="my_first_job", machine=Machine.A10G)
+s.installed_plugins["multi-machine-training"].run("python my_dummy_file", name="my_first_mmt", machine=Machine.T4, num_instances=42)
 
 print("Stopping Studio")
 s.stop()