lightning-sdk 0.1.57__py3-none-any.whl → 0.2.0__py3-none-any.whl

lightning_sdk/studio.py

@@ -160,7 +160,11 @@ class Studio:
         status = self.status
 
         if interruptible is None:
-            interruptible = self.teamspace.start_studions_on_interruptible
+            interruptible_override = os.environ.get("LIGHTNING_INTERRUPTIBLE_OVERRIDE", None)
+            if interruptible_override is not None:
+                interruptible = interruptible_override.lower() == "true"
+            else:
+                interruptible = self.teamspace.start_studios_on_interruptible
 
         if status == Status.Running:
             curr_machine = _machine_to_compute_name(self.machine) if self.machine is not None else None

lightning_sdk/teamspace.py

@@ -115,7 +115,7 @@ class Teamspace:
         return self._teamspace.project_settings.preferred_cluster
 
     @property
-    def start_studions_on_interruptible(self) -> bool:
+    def start_studios_on_interruptible(self) -> bool:
         return self._teamspace.project_settings.start_studio_on_spot_instance
 
     @property

lightning_sdk/utils/resolve.py

@@ -5,10 +5,12 @@ from contextlib import contextmanager
 from typing import TYPE_CHECKING, Generator, List, Optional, Tuple, Union
 
 from lightning_sdk.api import TeamspaceApi, UserApi
+from lightning_sdk.api.utils import _get_cloud_url
 from lightning_sdk.machine import Machine
 
 if TYPE_CHECKING:
     from lightning_sdk.organization import Organization
+    from lightning_sdk.studio import Studio
     from lightning_sdk.teamspace import Teamspace
     from lightning_sdk.user import User
 
@@ -179,7 +181,7 @@ def skip_studio_init() -> Generator[None, None, None]:
 def _parse_model_and_version(name: str) -> Tuple[str, str]:
     parts = name.split(":")
     if len(parts) == 1:
-        return parts[0], "latest"
+        return parts[0], "default"
     if len(parts) == 2:
         return parts[0], parts[1]
     # The rest of the validation for name and version happens in the backend
@@ -194,3 +196,12 @@ def in_studio() -> bool:
     has_cloudspace_id = bool(os.getenv("LIGHTNING_CLOUD_SPACE_ID", None))
     is_interactive = os.getenv("LIGHTNING_INTERACTIVE", "false") == "true"
     return has_cloudspace_id and is_interactive
+
+
+def _get_studio_url(studio: "Studio", turn_on: bool = False) -> str:
+    cloud_url = _get_cloud_url().replace(":443", "")
+    base_url = f"{cloud_url}/{studio.owner.name}/{studio.teamspace.name}/studios/{studio.name}/code"
+
+    if turn_on:
+        return f"{base_url}?turnOn=true"
+    return base_url

{lightning_sdk-0.1.57.dist-info → lightning_sdk-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning_sdk
-Version: 0.1.57
+Version: 0.2.0
 Summary: SDK to develop using Lightning AI Studios
 Author-email: Lightning-AI <justus@lightning.ai>
 License: MIT License
@@ -32,22 +32,20 @@ Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: backoff
+Requires-Dist: click
+Requires-Dist: docker
 Requires-Dist: fastapi
+Requires-Dist: packaging
 Requires-Dist: pyjwt
-Requires-Dist: python-multipart
 Requires-Dist: requests
 Requires-Dist: rich
+Requires-Dist: simple-term-menu
 Requires-Dist: six
+Requires-Dist: tqdm
 Requires-Dist: urllib3
 Requires-Dist: uvicorn
 Requires-Dist: websocket-client
-Requires-Dist: tqdm
-Requires-Dist: fire
-Requires-Dist: simple-term-menu
-Requires-Dist: lightning-utilities
-Requires-Dist: docker
 Requires-Dist: wget
-Requires-Dist: click
 Provides-Extra: serve
 Requires-Dist: litserve>=0.2.5; extra == "serve"