lightning-sdk 2025.12.17__py3-none-any.whl → 2026.1.22__py3-none-any.whl

lightning_sdk/studio.py

@@ -11,7 +11,7 @@ from lightning_sdk.api.studio_api import StudioApi
 from lightning_sdk.api.utils import AccessibleResource, raise_access_error_if_not_allowed
 from lightning_sdk.base_studio import BaseStudio
 from lightning_sdk.constants import _LIGHTNING_DEBUG
-from lightning_sdk.exceptions import OutOfCapacityError
+from lightning_sdk.exceptions import NotSupportedError, OutOfCapacityError
 from lightning_sdk.lightning_cloud.openapi import V1ClusterType, V1Endpoint
 from lightning_sdk.machine import DEFAULT_MACHINE, CloudProvider, Machine
 from lightning_sdk.organization import Organization
@@ -341,12 +341,20 @@ class Studio(metaclass=TrackCallsMeta):
             if new_machine != self.machine:
                 raise RuntimeError(
                     f"Requested to start {self._cls_name} on {new_machine}, "
-                    "but {self._cls_name} is already running on {self.machine}."
+                    f"but {self._cls_name} is already running on {self.machine}."
                     " Consider switching instead!"
                 )
             _logger.info(f"{self._cls_name} {self.name} is already running")
             return
 
+        if not self._studio_api.machine_is_supported(
+            new_machine, self._teamspace.id, self.cloud_account, _get_org_id(self._teamspace)
+        ):
+            raise NotSupportedError(
+                "Requested machine is not supported in the selected cloud account. "
+                "Try a different machine or cloud account by setting the `machine` or `cloud_account` argument."
+            )
+
         if not self._studio_api.machine_has_capacity(
             new_machine,
             self._teamspace.id,
@@ -548,7 +556,7 @@ class Studio(metaclass=TrackCallsMeta):
         if status != Status.Running:
             raise RuntimeError(
                 f"Cannot run a command in a {self._cls_name} that is not running. "
-                "{self._cls_name} {self.name} is {status}."
+                f"{self._cls_name} {self.name} is {status}."
             )
         output, exit_code = self._studio_api.run_studio_commands(self._studio.id, self._teamspace.id, *commands)
         output = output.strip()
@@ -605,11 +613,13 @@ class Studio(metaclass=TrackCallsMeta):
             progress_bar = tqdm(total=len(all_files), desc="Uploading files", unit="file")
         for local_file, remote_path in sorted(all_files, key=lambda p: p[1]):
             if progress_bar:
-                progress_bar.set_description(f"Uploading {local_file}")
+                progress_bar.set_description("Uploading files")
             self.upload_file(local_file, remote_path=remote_path, progress_bar=False)
             if progress_bar:
                 progress_bar.update(1)
         if progress_bar:
+            progress_bar.set_description("Upload complete")
+            progress_bar.refresh()
             progress_bar.close()
 
     def download_file(self, remote_path: str, file_path: Optional[str] = None) -> None:

lightning_sdk/utils/logging.py

@@ -22,7 +22,7 @@ def track_calls() -> Callable[..., any]:
             bound_args.apply_defaults()
 
             args_str = ", ".join(f"{k}: {v}" for k, v in bound_args.arguments.items() if k != "self")
-            message = f"VERSION: {__version__} | ARGS: {args_str} "
+            message = f"ARGS: {args_str} "
 
             body = V1CreateSDKCommandHistoryRequest(
                 command=func.__qualname__,
@@ -30,6 +30,7 @@ def track_calls() -> Callable[..., any]:
                 project_id=None,
                 severity=V1SDKCommandHistorySeverity.INFO,
                 type=V1SDKCommandHistoryType.SDK,
+                version=__version__,
             )
 
             try:

{lightning_sdk-2025.12.17.dist-info → lightning_sdk-2026.1.22.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning_sdk
-Version: 2025.12.17
+Version: 2026.1.22
 Summary: SDK to develop using Lightning AI Studios
 Author-email: Lightning-AI <justus@lightning.ai>
 License: MIT License
@@ -38,8 +38,6 @@ Requires-Dist: click
 Requires-Dist: docker
 Requires-Dist: fastapi
 Requires-Dist: packaging
-Requires-Dist: pandas
-Requires-Dist: pyjwt
 Requires-Dist: pyyaml
 Requires-Dist: requests
 Requires-Dist: rich
@@ -48,8 +46,6 @@ Requires-Dist: six
 Requires-Dist: tqdm
 Requires-Dist: urllib3<=2.5.0
 Requires-Dist: uvicorn
-Requires-Dist: websocket-client
-Requires-Dist: wget
 Provides-Extra: serve
 Requires-Dist: litserve>=0.2.5; extra == "serve"