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

lightning_sdk/machine.py

@@ -162,12 +162,12 @@ Machine.CPU_X_4 = Machine(name="CPU_X_4", slug="cpu-4", family="CPU", accelerato
 Machine.CPU_X_8 = Machine(name="CPU_X_8", slug="cpu-8", family="CPU", accelerator_count=8)
 Machine.CPU_X_16 = Machine(name="CPU_X_16", slug="cpu-16", family="CPU", accelerator_count=16)
 # available data-prep (big disk) machines
-Machine.DATA_PREP = Machine(name="DATA_PREP", slug="data-prep-mid", family="DATA_PREP", accelerator_count=32)
+Machine.DATA_PREP = Machine(name="DATA_PREP", slug="data-prep-mid", family="DATA-PREP", accelerator_count=32)
 Machine.DATA_PREP_MAX = Machine(
-    name="DATA_PREP_MAX", slug="data-prep-max-large", family="DATA_PREP", accelerator_count=64
+    name="DATA_PREP_MAX", slug="data-prep-max-large", family="DATA-PREP", accelerator_count=64
 )
 Machine.DATA_PREP_ULTRA = Machine(
-    name="DATA_PREP_ULTRA", slug="data-prep-ultra-extra-large", family="DATA_PREP", accelerator_count=96
+    name="DATA_PREP_ULTRA", slug="data-prep-ultra-extra-large", family="DATA-PREP", accelerator_count=96
 )
 
 # GPU machines

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/teamspace.py

@@ -539,21 +539,37 @@ class Teamspace(metaclass=TrackCallsMeta):
         return self._teamspace_api.list_model_versions(teamspace_id=self.id, model_name=name)
 
     def upload_file(
-        self, file_path: Union[Path, str], remote_path: Optional[str] = None, progress_bar: bool = True
+        self,
+        file_path: Union[Path, str],
+        remote_path: Optional[str] = None,
+        progress_bar: bool = True,
+        cloud_account: Optional[str] = None,
     ) -> None:
         """Uploads file to given remote path in the Teamspace drive."""
         if remote_path is None:
             remote_path = os.path.split(file_path)[1]
 
+        if cloud_account is None:
+            warnings.warn(
+                f"No cloud account specified. Using teamspace default cloud account: {self.default_cloud_account}."
+            )
+            cloud_account = self.default_cloud_account
+
         self._teamspace_api.upload_file(
             teamspace_id=self._teamspace.id,
-            cloud_account=self.default_cloud_account,
+            cloud_account=cloud_account,
             file_path=file_path,
             remote_path=os.path.normpath(remote_path),
             progress_bar=progress_bar,
         )
 
-    def upload_folder(self, folder_path: str, remote_path: Optional[str] = None, progress_bar: bool = True) -> None:
+    def upload_folder(
+        self,
+        folder_path: str,
+        remote_path: Optional[str] = None,
+        progress_bar: bool = True,
+        cloud_account: Optional[str] = None,
+    ) -> None:
         """Uploads a given folder to a remote path in the Teamspace drive."""
         if folder_path is None:
             raise ValueError("Cannot upload a folder that is None.")
@@ -575,13 +591,15 @@ class Teamspace(metaclass=TrackCallsMeta):
         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}")
-            self.upload_file(local_file, remote_path=remote_path, progress_bar=False)
+            self.upload_file(local_file, remote_path=remote_path, progress_bar=False, cloud_account=cloud_account)
             if progress_bar:
                 progress_bar.update(1)
         if progress_bar:
             progress_bar.close()
 
-    def download_file(self, remote_path: str, file_path: Optional[str] = None) -> None:
+    def download_file(
+        self, remote_path: str, file_path: Optional[str] = None, cloud_account: Optional[str] = None
+    ) -> None:
         """Downloads a given file in Teamspace drive to a target location."""
         if file_path is None:
             file_path = remote_path
@@ -590,9 +608,12 @@ class Teamspace(metaclass=TrackCallsMeta):
             path=remote_path,
             target_path=file_path,
             teamspace_id=self._teamspace.id,
+            cloud_account=cloud_account,
         )
 
-    def download_folder(self, remote_path: str, target_path: Optional[str] = None) -> None:
+    def download_folder(
+        self, remote_path: str, target_path: Optional[str] = None, cloud_account: Optional[str] = None
+    ) -> None:
         """Downloads a folder in the Teamspace drive to a given target path."""
         if target_path is None:
             target_path = remote_path
@@ -601,7 +622,7 @@ class Teamspace(metaclass=TrackCallsMeta):
             path=remote_path,
             target_path=target_path,
             teamspace_id=self._teamspace.id,
-            cloud_account=self.default_cloud_account,
+            cloud_account=cloud_account,
         )
 
     def new_folder(

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.27.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning_sdk
-Version: 2025.12.17
+Version: 2026.1.27
 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"