lightning-sdk 2025.10.31__py3-none-any.whl → 2025.11.13__py3-none-any.whl

lightning_sdk/__init__.py

@@ -1,8 +1,9 @@
+from lightning_sdk.__version__ import __version__
 from lightning_sdk.agents import Agent
 from lightning_sdk.ai_hub import AIHub
 from lightning_sdk.constants import __GLOBAL_LIGHTNING_UNIQUE_IDS_STORE__  # noqa: F401
 from lightning_sdk.deployment import Deployment
-from lightning_sdk.helpers import VersionChecker, _set_tqdm_envvars_noninteractive
+from lightning_sdk.helpers import VersionChecker, set_tqdm_envvars_noninteractive
 from lightning_sdk.job import Job
 from lightning_sdk.machine import CloudProvider, Machine
 from lightning_sdk.mmt import MMT
@@ -14,6 +15,8 @@ from lightning_sdk.teamspace import ConnectionType, FolderLocation, Teamspace
 from lightning_sdk.user import User
 
 __all__ = [
+    "MMT",
+    "VM",
     "AIHub",
     "Agent",
     "CloudProvider",
@@ -23,7 +26,6 @@ __all__ = [
     "Job",
     "JobsPlugin",
     "Machine",
-    "MMT",
     "MultiMachineTrainingPlugin",
     "Organization",
     "Plugin",
@@ -32,12 +34,10 @@ __all__ = [
     "Studio",
     "Teamspace",
     "User",
-    "VM",
+    "__version__",
 ]
 
-__version__ = "2025.10.31"
-
 _version_checker = VersionChecker()
 _version_checker.check_and_prompt_upgrade(__version__)
 
-_set_tqdm_envvars_noninteractive()
+set_tqdm_envvars_noninteractive()

lightning_sdk/__version__.py

@@ -0,0 +1,3 @@
+"""Version information for lightning_sdk."""
+
+__version__ = "2025.11.13"

lightning_sdk/agents.py

@@ -1,9 +1,10 @@
 from typing import List, Optional
 
 from lightning_sdk.api.agents_api import AgentApi
+from lightning_sdk.utils.logging import TrackCallsMeta
 
 
-class Agent:
+class Agent(metaclass=TrackCallsMeta):
     def __init__(self, agent_id: str) -> None:
         self.id = agent_id
         self._agent_api = AgentApi()

lightning_sdk/ai_hub.py

@@ -5,6 +5,7 @@ from lightning_sdk.api import AIHubApi
 from lightning_sdk.api.utils import _get_cloud_url
 from lightning_sdk.lightning_cloud.openapi.models import V1Deployment
 from lightning_sdk.user import User
+from lightning_sdk.utils.logging import TrackCallsMeta
 from lightning_sdk.utils.resolve import _resolve_teamspace
 
 if TYPE_CHECKING:
@@ -12,7 +13,7 @@ if TYPE_CHECKING:
     from lightning_sdk.machine import Machine
 
 
-class AIHub:
+class AIHub(metaclass=TrackCallsMeta):
     """An interface to interact with the AI Hub.
 
     Example:

lightning_sdk/api/cloud_account_api.py

@@ -198,8 +198,8 @@ class CloudAccountApi:
                 cloud_provider_resp = self._get_cloud_account_provider(cloud_account_resp)
                 if cloud_provider_resp != cloud_provider:
                     raise RuntimeError(
-                        f"Specified both cloud_provider ({cloud_provider}) and "
-                        "cloud_account ({cloud_account} has cloud provider {cloud_provider_resp}) which don't match!"
+                        f"Specified both cloud_provider ({cloud_provider}) and cloud_account ({cloud_account} "
+                        f"has cloud provider {cloud_provider_resp}) which don't match!"
                     )
 
             return cloud_account

lightning_sdk/api/deployment_api.py

@@ -1,7 +1,7 @@
 from time import sleep
 from typing import Any, Dict, List, Literal, Optional, Union
 
-from lightning_sdk.api.utils import _machine_to_compute_name
+from lightning_sdk.api.utils import _machine_to_compute_name, resolve_path_mappings
 from lightning_sdk.lightning_cloud.openapi import (
     CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs,
     V1AutoscalingSpec,
@@ -269,6 +269,7 @@ class DeploymentApi:
         quantity: Optional[int] = None,
         include_credentials: Optional[bool] = None,
         max_runtime: Optional[int] = None,
+        path_mappings: Optional[Dict[str, str]] = None,
     ) -> V1Deployment:
         # Update the deployment in place
 
@@ -288,6 +289,11 @@ class DeploymentApi:
         requires_release = False
         requires_release |= apply_change(deployment.spec, "image", image)
 
+        if path_mappings:
+            requires_release |= apply_change(
+                deployment.spec, "path_mappings", resolve_path_mappings(path_mappings, None, None)
+            )
+
         requires_release |= apply_change(deployment.spec, "entrypoint", entrypoint)
         requires_release |= apply_change(deployment.spec, "command", command)
         requires_release |= apply_change(deployment.spec, "env", to_env(env))
@@ -591,6 +597,7 @@ def to_spec(
     cloudspace_id: Optional[None] = None,
     max_runtime: Optional[int] = None,
     machine_image_version: Optional[str] = None,
+    path_mappings: Optional[Dict[str, str]] = None,
 ) -> V1JobSpec:
     if cloud_account is None:
         raise ValueError("The cloud account should be defined.")
@@ -612,6 +619,8 @@ def to_spec(
     if max_runtime:
         optional_spec_kwargs["requested_run_duration_seconds"] = str(max_runtime)
 
+    path_mapping_list = resolve_path_mappings(path_mappings or {}, None, None)
+
     return V1JobSpec(
         cluster_id=cloud_account,
         command=command,
@@ -625,6 +634,7 @@ def to_spec(
         include_credentials=include_credentials,
         cloudspace_id=cloudspace_id,
         machine_image_version=machine_image_version,
+        path_mappings=path_mapping_list,
         **optional_spec_kwargs,
     )
 

lightning_sdk/api/utils.py

@@ -4,7 +4,8 @@ import math
 import os
 import re
 from concurrent.futures import ThreadPoolExecutor
-from functools import partial
+from enum import Enum
+from functools import lru_cache, partial
 from pathlib import Path
 from typing import Any, Callable, Dict, List, Optional, Tuple, TypedDict, Union
 
@@ -747,3 +748,59 @@ def resolve_path_mappings(
             )
 
     return path_mappings_list
+
+
+class AccessibleResource(Enum):
+    Studios = "studio"
+    Drive = "drive"
+    Jobs = "jobs"
+    Deployments = "deployments"
+    Pipelines = "pipelines"
+    Models = "models"
+    Containers = "containers"
+    Settings = "settings"
+
+    def __str__(self) -> str:
+        """Return the string representation of the resource type."""
+        return self.value
+
+    def __repr__(self) -> str:
+        """Return the string representation of the resource type."""
+        return self.value
+
+    def __eq__(self, other: object) -> bool:
+        """Return True if the resource type is equal to the other resource type."""
+        if isinstance(other, AccessibleResource):
+            return self.value == other.value
+        return str(other) == self.value
+
+    def __hash__(self) -> int:
+        """Return the hash of the resource type."""
+        return hash(self.value)
+
+
+@lru_cache
+def allowed_resource_access(resource_type: AccessibleResource, teamspace_id: str) -> bool:
+    # TODO: change this to proper API
+    from lightning_sdk.api.teamspace_api import TeamspaceApi
+
+    teamspace_api = TeamspaceApi()
+    teamspace = teamspace_api._get_teamspace_by_id(teamspace_id=teamspace_id)
+
+    # when we find the tab, check if it is enabled
+    if teamspace.layout_config:
+        for tab in teamspace.layout_config:
+            if tab.slug == resource_type:
+                return tab.is_enabled
+
+    # tab isn't found, allow access by default for backwards compatibility
+    # TODO: add additional checks here if required
+    return True
+
+
+def raise_access_error_if_not_allowed(resource_type: AccessibleResource, teamspace_id: str) -> None:
+    if not allowed_resource_access(resource_type, teamspace_id):
+        raise PermissionError(
+            f"Access to {resource_type.name} has been disabled for this teamspace. "
+            "Contact a teamspace administrator to enable it."
+        )

lightning_sdk/cli/legacy/deploy/serve.py

@@ -12,7 +12,7 @@ from rich.console import Console
 from rich.progress import Progress, SpinnerColumn, TextColumn, TimeElapsedColumn
 from rich.prompt import Confirm
 
-from lightning_sdk import Machine, Teamspace
+from lightning_sdk import CloudProvider, Machine, Teamspace
 from lightning_sdk.api.lit_container_api import LitContainerApi
 from lightning_sdk.api.utils import _get_registry_url
 from lightning_sdk.cli.legacy.clusters_menu import _ClustersMenu
@@ -122,6 +122,12 @@ def deploy() -> None:
         "If not provided will fall back to the teamspaces default cloud account."
     ),
 )
+@click.option(
+    "--cloud-provider",
+    "--cloud_provider",
+    default=None,
+    help="The provider to create the studio on. If --cloud-account is specified, this option is prioritized.",
+)
 @click.option("--port", default=8000, help="The port to expose the API on.")
 @click.option("--min_replica", "--min-replica", default=0, help="Number of replicas to start with.")
 @click.option("--max_replica", "--max-replica", default=1, help="Number of replicas to scale up to.")
@@ -152,6 +158,7 @@ def api(
     max_replica: Optional[int],
     replicas: Optional[int],
     no_credentials: Optional[bool],
+    cloud_provider: Optional[str],
 ) -> None:
     """Deploy a LitServe model script."""
     return api_impl(
@@ -172,6 +179,7 @@ def api(
         min_replica=min_replica,
         max_replica=max_replica,
         include_credentials=not no_credentials,
+        cloud_provider=cloud_provider,
     )
 
 
@@ -194,6 +202,7 @@ def api_impl(
     max_replica: Optional[int] = 1,
     replicas: Optional[int] = 1,
     include_credentials: Optional[bool] = True,
+    cloud_provider: Optional[str] = None,
 ) -> None:
     """Deploy a LitServe model script."""
     console = Console()
@@ -226,6 +235,7 @@ def api_impl(
         return _handle_devbox(name, script_path, console, non_interactive, machine, interruptible, teamspace, org, user)
 
     machine = Machine.from_str(machine)
+    cloud_provider = CloudProvider.from_str(cloud_provider) if cloud_provider else None
     return _handle_cloud(
         script_path,
         console,
@@ -243,6 +253,7 @@ def api_impl(
         max_replica=max_replica,
         replicas=replicas,
         include_credentials=include_credentials,
+        cloud_provider=cloud_provider,
     )
 
 
@@ -304,6 +315,7 @@ def _handle_cloud(
     max_replica: Optional[int] = 1,
     replicas: Optional[int] = 1,
     include_credentials: Optional[bool] = True,
+    cloud_provider: Optional[CloudProvider] = None,
 ) -> None:
     if not is_connected():
         console.print("❌ Internet connection required to deploy to the cloud.", style="red")
@@ -379,7 +391,7 @@ def _handle_cloud(
         resolved_teamspace = select_teamspace(teamspace, org, user)
 
     lightning_containers_cloud_account = cloud_account
-    if not cloud_account:
+    if not cloud_account and not cloud_provider:
         clusters_menu = _ClustersMenu()
         lightning_containers_cloud_account = clusters_menu._resolve_cluster(resolved_teamspace)
         cloud_account = resolved_teamspace.default_cloud_account
@@ -414,6 +426,7 @@ def _handle_cloud(
                 "include_credentials": include_credentials,
                 "cloudspace_id": cloudspace_id,
                 "from_onboarding": from_onboarding,
+                "cloud_provider": cloud_provider,
             },
         )
         thread.start()
@@ -446,6 +459,7 @@ def _handle_cloud(
         include_credentials=include_credentials,
         cloudspace_id=cloudspace_id,
         from_onboarding=from_onboarding,
+        cloud_provider=cloud_provider,
     )
     console.print(f"🚀 Deployment started, access at [i]{deployment_status.get('url')}[/i]")
     if user_status["onboarded"]:

lightning_sdk/cli/studio/__init__.py

@@ -6,6 +6,7 @@ import click
 def register_commands(group: click.Group) -> None:
     """Register studio commands with the given group."""
     from lightning_sdk.cli.studio.connect import connect_studio
+    from lightning_sdk.cli.studio.cp import cp_studio_file
     from lightning_sdk.cli.studio.create import create_studio
     from lightning_sdk.cli.studio.delete import delete_studio
     from lightning_sdk.cli.studio.list import list_studios
@@ -22,3 +23,4 @@ def register_commands(group: click.Group) -> None:
     group.add_command(stop_studio)
     group.add_command(switch_studio)
     group.add_command(connect_studio)
+    group.add_command(cp_studio_file)

lightning_sdk/cli/studio/cp.py

@@ -0,0 +1,138 @@
+"""Studio cp command."""
+
+from pathlib import Path
+from typing import Optional, TypedDict
+
+import click
+from rich.console import Console
+
+from lightning_sdk.api.utils import _get_cloud_url
+from lightning_sdk.cli.legacy.exceptions import StudioCliError
+from lightning_sdk.cli.utils.owner_selection import OwnerMenu
+from lightning_sdk.cli.utils.studio_selection import StudiosMenu
+from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
+from lightning_sdk.studio import Studio
+
+
+@click.command("cp")
+@click.argument("source", nargs=1)
+@click.argument("destination", nargs=1)
+def cp_studio_file(source: str, destination: str, teamspace: Optional[str] = None) -> None:
+    """Copy a Studio file.
+
+    SOURCE: Source file to copy from. For Studio files, use the format lit://<owner>/<my-teamspace>/studios/<my-studio>/<filepath>.
+
+    DESTINATION: Destination file to copy to. For Studio files, use the format lit://<owner>/<my-teamspace>/studios/<my-studio>/<filepath>.
+
+    Example:
+        lightning studio cp source.txt lit://<owner>/<my-teamspace>/studios/<my-studio>/destination.txt
+
+    """
+    return cp_impl(source=source, destination=destination)
+
+
+def cp_impl(source: str, destination: str) -> None:
+    if "lit://" in source and "lit://" in destination:
+        raise ValueError("Both source and destination cannot be Studio files.")
+    elif "lit://" not in source and "lit://" not in destination:
+        raise ValueError("Either source or destination must be a Studio file.")
+    elif "lit://" in source:
+        # Download from Studio to local
+        cp_download(studio_file_path=source, local_file_path=destination)
+    else:
+        # Upload from local to Studio
+        cp_upload(local_file_path=source, studio_file_path=destination)
+
+
+class StudioPathResult(TypedDict):
+    owner: Optional[str]
+    teamspace: Optional[str]
+    studio: Optional[str]
+    destination: Optional[str]
+
+
+def parse_studio_path(studio_path: str) -> StudioPathResult:
+    path_string = studio_path.removeprefix("lit://")
+    if not path_string:
+        raise ValueError("Studio path cannot be empty after prefix")
+
+    result: StudioPathResult = {"owner": None, "teamspace": None, "studio": None, "destination": None}
+
+    if "/studios/" in path_string:
+        prefix_part, suffix_part = path_string.split("/studios/", 1)
+
+        # org and teamspace
+        if prefix_part:
+            org_ts_components = prefix_part.split("/")
+            if len(org_ts_components) == 2:
+                result["owner"], result["teamspace"] = org_ts_components
+            elif len(org_ts_components) == 1:
+                result["teamspace"] = org_ts_components[0]
+            else:
+                raise ValueError(f"Invalid format: '{prefix_part}'")
+
+        # studio and destination
+        path_parts = suffix_part.split("/")
+
+    else:
+        # studio and destination
+        path_parts = path_string.split("/")
+
+    if not path_parts or len(path_parts) < 2:
+        raise ValueError("Invalid: Missing studio name.")
+
+    result["studio"] = path_parts[0]
+    result["destination"] = "/".join(path_parts[1:])
+
+    return result
+
+
+def cp_upload(
+    local_file_path: str,
+    studio_file_path: str,
+) -> None:
+    console = Console()
+    if Path(local_file_path).is_dir():
+        raise StudioCliError(
+            f"The provided path is a folder: {local_file_path}. Use `lightning upload folder` instead."
+        )
+    if not Path(local_file_path).exists():
+        raise FileNotFoundError(f"The provided path does not exist: {local_file_path}.")
+
+    studio_path_result = parse_studio_path(studio_file_path)
+
+    selected_studio = resolve_studio(
+        studio_path_result["studio"], studio_path_result["teamspace"], studio_path_result["owner"]
+    )
+    console.print(f"Uploading to {selected_studio.teamspace.name}/{selected_studio.name}")
+
+    selected_studio.upload_file(local_file_path, studio_file_path)
+
+    studio_url = (
+        _get_cloud_url().replace(":443", "")
+        + "/"
+        + selected_studio.owner.name
+        + "/"
+        + selected_studio.teamspace.name
+        + "/studios/"
+        + selected_studio.name
+    )
+    console.print(f"See your file at {studio_url}")
+
+
+def cp_download(
+    studio_file_path: str,
+    local_file_path: str,
+) -> None:
+    raise NotImplementedError("Download functionality is not implemented yet.")
+
+
+def resolve_studio(studio_name: Optional[str], teamspace: Optional[str], owner: Optional[str]) -> Studio:
+    owner_menu = OwnerMenu()
+    resolved_owner = owner_menu(owner=owner)
+
+    teamspace_menu = TeamspacesMenu(resolved_owner)
+    resolved_teamspace = teamspace_menu(teamspace=teamspace)
+
+    studio_menu = StudiosMenu(resolved_teamspace)
+    return studio_menu(studio=studio_name)

lightning_sdk/cli/utils/logging.py

@@ -12,6 +12,7 @@ from rich.panel import Panel
 from rich.syntax import Syntax
 from rich.text import Text
 
+from lightning_sdk.__version__ import __version__
 from lightning_sdk.cli.utils import rich_to_str
 from lightning_sdk.constants import _LIGHTNING_DEBUG
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_sdk_command_history_request import (
@@ -29,7 +30,7 @@ def _log_command(message: str = "", duration: int = 0, error: Optional[str] = No
     body = V1CreateSDKCommandHistoryRequest(
         command=original_command,
         duration=duration,
-        message=message,
+        message=f"VERSION: {__version__} | {message}",
         project_id=None,
         severity=V1SDKCommandHistorySeverity.INFO,
         type=V1SDKCommandHistoryType.CLI,

lightning_sdk/cli/utils/studio_selection.py

@@ -52,11 +52,11 @@ class StudiosMenu:
 
     def _get_possible_studios(self) -> Dict[str, Union[Studio, VM]]:
         """Get all available studios in the teamspace."""
-        studios = {}
+        studios: Dict[str, Union[Studio, VM]] = {}
 
         user = _get_authed_user()
-        studios = self.teamspace.vms if self.vm else self.teamspace.studios
-        for studio in self.teamspace.studios:
+        teamspace_studios = self.teamspace.vms if self.vm else self.teamspace.studios
+        for studio in teamspace_studios:
             if studio._studio.user_id == user.id:
                 studios[studio.name] = studio
         return studios

lightning_sdk/deployment/__init__.py

@@ -1,4 +1,5 @@
 from lightning_sdk.api.deployment_api import (
+    ApiKeyAuth,
     AutoScaleConfig,
     AutoScalingMetric,
     BasicAuth,
@@ -24,4 +25,5 @@ __all__ = [
     "Secret",
     "TokenAuth",
     "Deployment",
+    "ApiKeyAuth",
 ]

lightning_sdk/deployment/deployment.py

@@ -4,7 +4,7 @@ from typing import Any, Dict, List, Optional, Union
 
 import requests
 
-from lightning_sdk.api import UserApi
+from lightning_sdk.api import CloudAccountApi, UserApi
 from lightning_sdk.api.deployment_api import (
     ApiKeyAuth,
     Auth,
@@ -28,18 +28,20 @@ from lightning_sdk.api.deployment_api import (
     to_spec,
     to_strategy,
 )
+from lightning_sdk.api.utils import AccessibleResource, raise_access_error_if_not_allowed
 from lightning_sdk.lightning_cloud import login
 from lightning_sdk.lightning_cloud.openapi import V1Deployment
-from lightning_sdk.machine import Machine
+from lightning_sdk.machine import CloudProvider, Machine
 from lightning_sdk.organization import Organization
 from lightning_sdk.services.utilities import _get_cluster
 from lightning_sdk.studio import Studio
 from lightning_sdk.teamspace import Teamspace
 from lightning_sdk.user import User
+from lightning_sdk.utils.logging import TrackCallsMeta
 from lightning_sdk.utils.resolve import _resolve_deprecated_cluster, _resolve_org, _resolve_teamspace, _resolve_user
 
 
-class Deployment:
+class Deployment(metaclass=TrackCallsMeta):
     """The Lightning AI Deployment.
 
     Allows to fully control a deployment, including retrieving the status, making new release
@@ -65,6 +67,7 @@ class Deployment:
         user: Optional[Union[str, User]] = None,
     ) -> None:
         self._request_session = None
+        self._cloud_account_api = CloudAccountApi()
 
         self._auth = login.Auth()
         self._user = None
@@ -91,6 +94,8 @@ class Deployment:
         if self._teamspace is None:
             raise ValueError("You need to pass a teamspace or an org for your deployment.")
 
+        raise_access_error_if_not_allowed(AccessibleResource.Deployments, self._teamspace.id)
+
         self._deployment_api = DeploymentApi()
         self._cloud_account = _get_cluster(client=self._deployment_api._client, project_id=self._teamspace.id)
         self._is_created = False
@@ -132,6 +137,8 @@ class Deployment:
         from_onboarding: Optional[bool] = None,
         from_litserve: Optional[bool] = None,
         max_runtime: Optional[int] = None,
+        path_mappings: Optional[Dict[str, str]] = None,
+        cloud_provider: Optional[CloudProvider] = None,
     ) -> None:
         """The Lightning AI Deployment.
 
@@ -163,12 +170,22 @@ class Deployment:
                 Irrelevant for most machines, required for some of the top-end machines on GCP.
                 If in doubt, set it. Won't have an effect on machines not requiring it.
                 Defaults to 3h
+            path_mappings: Dictionary of path mappings. The keys are the path inside the container whereas the value
+                represents the data-connection name and the path inside that connection.
+                Should be of form
+                    {
+                        "<CONTAINER_PATH_1>": "<CONNECTION_NAME_1>:<PATH_WITHIN_CONNECTION_1>",
+                        "<CONTAINER_PATH_2>": "<CONNECTION_NAME_2>"
+                    }
+                If the path inside the connection is omitted it's assumed to be the root path of that connection.
+                Only applicable when deploying docker containers.
 
         Note:
             Since a teamspace can either be owned by an org or by a user directly,
             only one of the arguments can be provided.
 
         """
+        raise_access_error_if_not_allowed(AccessibleResource.Deployments, self._teamspace.id)
         if self._is_created:
             raise RuntimeError("This deployment has already been started.")
 
@@ -188,10 +205,17 @@ class Deployment:
         if cloud_account is None:
             cloud_account = _resolve_deprecated_cluster(cloud_account, cluster)
 
-        if cloud_account is None and self._cloud_account is not None:
+        if cloud_account is None and self._cloud_account is not None and cloud_provider is None:
             print(f"No cloud account was provided, defaulting to {self._cloud_account.cluster_id}")
             cloud_account = os.getenv("LIGHTNING_CLUSTER_ID") or self._cloud_account.cluster_id
 
+        _cloud_account = self._cloud_account_api.resolve_cloud_account(
+            self.teamspace.id,
+            cloud_account=cloud_account,
+            cloud_provider=cloud_provider,
+            default_cloud_account=self._teamspace.default_cloud_account,
+        )
+
         if isinstance(ports, float):
             ports = [ports]
 
@@ -226,7 +250,7 @@ class Deployment:
                 replicas=replicas,
                 cloudspace_id=cloudspace_id,
                 spec=to_spec(
-                    cloud_account=cloud_account,
+                    cloud_account=_cloud_account,
                     command=command,
                     entrypoint=entrypoint,
                     env=env,
@@ -239,6 +263,7 @@ class Deployment:
                     cloudspace_id=cloudspace_id,
                     max_runtime=max_runtime,
                     machine_image_version=machine_image_version,
+                    path_mappings=path_mappings,
                 ),
                 strategy=to_strategy(release_strategy),
             ),
@@ -275,7 +300,9 @@ class Deployment:
         quantity: Optional[int] = None,
         include_credentials: Optional[bool] = None,
         max_runtime: Optional[int] = None,
+        path_mappings: Optional[Dict[str, str]] = None,
     ) -> None:
+        raise_access_error_if_not_allowed(AccessibleResource.Deployments, self._teamspace.id)
         cloud_account = _resolve_deprecated_cluster(cloud_account, cluster)
 
         if command is None and commands is not None:
@@ -302,6 +329,7 @@ class Deployment:
             quantity=quantity,
             include_credentials=include_credentials,
             max_runtime=max_runtime,
+            path_mappings=path_mappings,
         )
 
     def stop(self) -> None:

lightning_sdk/helpers.py

@@ -73,7 +73,7 @@ class VersionChecker:
             self._warning_shown = True
 
 
-def _set_tqdm_envvars_noninteractive() -> None:
+def set_tqdm_envvars_noninteractive() -> None:
     # note: stderr is the default stream tqdm writes progressbars to
     # so we check that one.
     if os.isatty(sys.stderr.fileno()):

lightning_sdk/job/base.py

@@ -4,6 +4,7 @@ from typing import TYPE_CHECKING, Any, Dict, Optional, TypedDict, Union
 
 from lightning_sdk.api.cloud_account_api import CloudAccountApi
 from lightning_sdk.api.utils import _get_cloud_url
+from lightning_sdk.utils.logging import TrackCallsABCMeta
 from lightning_sdk.utils.resolve import _resolve_deprecated_cluster, _resolve_teamspace, in_studio, skip_studio_setup
 
 if TYPE_CHECKING:
@@ -29,7 +30,7 @@ class JobDict(MachineDict):
     total_cost: float
 
 
-class _BaseJob(ABC):
+class _BaseJob(ABC, metaclass=TrackCallsABCMeta):
     """Base interface to all job types."""
 
     def __init__(