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

lightning_sdk/cli/groups.py

@@ -4,6 +4,7 @@ import click
 
 from lightning_sdk.cli.base_studio import register_commands as register_base_studio_commands
 from lightning_sdk.cli.config import register_commands as register_config_commands
+from lightning_sdk.cli.cp import register_commands as register_cp_commands
 from lightning_sdk.cli.job import register_commands as register_job_commands
 from lightning_sdk.cli.license import register_commands as register_license_commands
 from lightning_sdk.cli.mmt import register_commands as register_mmt_commands
@@ -46,6 +47,26 @@ def license() -> None:  # noqa: A001
     """Manage Lightning AI Product Licenses."""
 
 
+@click.command(name="cp")
+@click.argument("source")
+@click.argument("destination", required=False)
+@click.option("--recursive", "-r", is_flag=True, help="Copy directories recursively")
+@click.pass_context
+def cp() -> None:
+    """Copy files between local filesystem, Studios, and teamspace drives.
+
+    \b
+    URL formats:
+      Studios:          lit://<owner>/<teamspace>/studios/<studio-name>/<path>
+      Teamspace drives: lit://<owner>/<teamspace>/uploads/<path>
+
+    \b
+    Examples:
+      lightning studio cp source.txt lit://<owner>/<my-teamspace>/studios/<my-studio>/destination.txt
+      lightning studio cp -r source_folder/ lit://<owner>/<my-teamspace>/studios/<my-studio>/destination_folder/
+    """
+
+
 # Register config commands with the main config group
 register_job_commands(job)
 register_mmt_commands(mmt)
@@ -54,3 +75,4 @@ register_config_commands(config)
 register_vm_commands(vm)
 register_base_studio_commands(base_studio)
 register_license_commands(license)
+register_cp_commands(cp)

lightning_sdk/cli/legacy/clusters_menu.py

@@ -32,8 +32,8 @@ class _ClustersMenu:
 
             cloud_account_api = CloudAccountApi()
 
-            resolved_cluster_obj = cloud_account_api.get_cloud_account(
-                cloud_account_id=selected_cluster_id, org_id=teamspace.owner.id, teamspace_id=teamspace.id
+            resolved_cluster_obj = cloud_account_api.get_cloud_account_non_org(
+                cloud_account_id=selected_cluster_id, teamspace_id=teamspace.id
             )
 
             return None if resolved_cluster_obj.spec.cluster_type == V1ClusterType.GLOBAL else resolved_cluster_obj.id

lightning_sdk/cli/legacy/deploy/_auth.py

@@ -8,7 +8,7 @@ from urllib.parse import urlencode
 from rich.console import Console
 from rich.prompt import Confirm
 
-from lightning_sdk import Teamspace
+from lightning_sdk import Organization, Teamspace
 from lightning_sdk.api import UserApi
 from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
 from lightning_sdk.lightning_cloud import env
@@ -74,13 +74,14 @@ def authenticate(mode: _AuthMode, shall_confirm: bool = True) -> None:
 def select_teamspace(teamspace: Optional[str], org: Optional[str], user: Optional[str]) -> Teamspace:
     if teamspace is None:
         menu = TeamspacesMenu()
-        possible_teamspaces = menu._get_possible_teamspaces(_get_authed_user())
+        auth_user = _get_authed_user()
+        possible_teamspaces = menu._get_possible_teamspaces(auth_user)
         if len(possible_teamspaces) == 1:
-            name = next(iter(possible_teamspaces.values()))["name"]
-            return Teamspace(name=name, org=org, user=user)
-
+            teamspace_name = next(iter(possible_teamspaces.values()))
+            if isinstance(menu._owner, Organization):
+                return Teamspace(name=teamspace_name, org=menu._owner, user=None)
+            return Teamspace(name=teamspace_name, org=None, user=menu._owner)
         return menu(teamspace)
-
     return _resolve_teamspace(teamspace=teamspace, org=org, user=user)
 
 

lightning_sdk/cli/legacy/download.py

@@ -242,7 +242,8 @@ def _resolve_studio(studio: Optional[str]) -> Studio:
                     and (re.match(studio_name, st["name"]) or studio_name in st["name"]),
                     possible_studios,
                 )
-                if not len(possible_studios):
+                possible_studios = list(possible_studios)
+                if len(possible_studios) == 0:
                     raise ValueError(
                         f"Could not find Studio like '{studio}', please consider update your filtering pattern."
                     )

lightning_sdk/cli/legacy/run.py

@@ -50,6 +50,15 @@ def run() -> None:
     help="The organization owning the teamspace (if any). Defaults to the current organization.",
 )
 @click.option("--user", default=None, help="The user owning the teamspace (if any). Defaults to the current user.")
+@click.option(
+    "--cloud-provider",
+    "--cloud_provider",
+    default=None,
+    help=(
+        "The provider to create the studio on. If set, must be in agreement with the provider from the "
+        "cloud_account (if specified)."
+    ),
+)
 @click.option(
     "--cloud-account",
     "--cloud_account",
@@ -96,7 +105,7 @@ def run() -> None:
 )
 @click.option(
     "--entrypoint",
-    default="sh -c",
+    default=None,
     show_default=True,
     help=(
         "The entrypoint of your docker container. "
@@ -141,6 +150,7 @@ def job(
     teamspace: Optional[str] = None,
     org: Optional[str] = None,
     user: Optional[str] = None,
+    cloud_provider: Optional[str] = None,
     cloud_account: Optional[str] = None,
     env: Sequence[str] = (),
     interruptible: bool = False,
@@ -184,6 +194,7 @@ def job(
         teamspace=resolved_teamspace,
         org=org,
         user=user,
+        cloud_provider=cloud_provider,
         cloud_account=cloud_account,
         env=env_dict,
         interruptible=interruptible,
@@ -289,7 +300,7 @@ def job(
 )
 @click.option(
     "--entrypoint",
-    default="sh -c",
+    default=None,
     show_default=True,
     help=(
         "The entrypoint of your docker container. "

lightning_sdk/cli/legacy/studios_menu.py

@@ -31,7 +31,14 @@ class _StudiosMenu:
             title = "Please select a Studio of the following studios:"
 
         return TerminalMenu(
-            [f"{s['teamspace']}/{s['name']}" for s in possible_studios], title=title, clear_menu_on_exit=True
+            [
+                f"{s['org']}/{s['teamspace']}/{s['name']}"
+                if s.get("org") is not None
+                else f"{s['teamspace']}/{s['name']}"
+                for s in possible_studios
+            ],
+            title=title,
+            clear_menu_on_exit=True,
         )
 
     @staticmethod

lightning_sdk/cli/studio/__init__.py

@@ -10,6 +10,8 @@ def register_commands(group: click.Group) -> None:
     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
+    from lightning_sdk.cli.studio.ls import ls_studio
+    from lightning_sdk.cli.studio.rm import rm_studio_file
     from lightning_sdk.cli.studio.ssh import ssh_studio
     from lightning_sdk.cli.studio.start import start_studio
     from lightning_sdk.cli.studio.stop import stop_studio
@@ -24,3 +26,5 @@ def register_commands(group: click.Group) -> None:
     group.add_command(switch_studio)
     group.add_command(connect_studio)
     group.add_command(cp_studio_file)
+    group.add_command(ls_studio)
+    group.add_command(rm_studio_file)

lightning_sdk/cli/studio/cp.py

@@ -2,22 +2,20 @@
 
 import os
 from pathlib import Path
-from typing import Optional, TypedDict
+from typing import Optional
 
 import click
 from rich.console import Console
 
 from lightning_sdk.api.utils import _get_cloud_url
-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
+from lightning_sdk.cli.utils.studio_filesystem import parse_studio_path, resolve_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:
+@click.option("-r", "--recursive", is_flag=True, help="Copy directories recursively")
+def cp_studio_file(source: str, destination: str, teamspace: Optional[str] = None, recursive: bool = False) -> 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>.
@@ -26,70 +24,29 @@ def cp_studio_file(source: str, destination: str, teamspace: Optional[str] = Non
 
     Example:
         lightning studio cp source.txt lit://<owner>/<my-teamspace>/studios/<my-studio>/destination.txt
+        lightning studio cp -r source_folder/ lit://<owner>/<my-teamspace>/studios/<my-studio>/destination_folder/
 
     """
-    return cp_impl(source=source, destination=destination)
+    return cp_impl(source=source, destination=destination, recursive=recursive)
 
 
-def cp_impl(source: str, destination: str) -> None:
+def cp_impl(source: str, destination: str, recursive: bool = False) -> 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_path=source, local_path=destination)
+        cp_download(studio_path=source, local_path=destination, recursive=recursive)
     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
+        cp_upload(local_file_path=source, studio_file_path=destination, recursive=recursive)
 
 
 def cp_upload(
     local_file_path: str,
     studio_file_path: str,
+    recursive: bool = False,
 ) -> None:
     console = Console()
     if not Path(local_file_path).exists():
@@ -103,6 +60,8 @@ def cp_upload(
     console.print(f"Uploading to {selected_studio.teamspace.name}/{selected_studio.name}")
 
     if Path(local_file_path).is_dir():
+        if not recursive:
+            raise ValueError(f"'{local_file_path}' is a directory. Use -r flag to copy directories recursively.")
         selected_studio.upload_folder(local_file_path, studio_path_result["destination"])
     else:
         if studio_file_path.endswith(("/", "\\")):
@@ -126,6 +85,7 @@ def cp_upload(
 def cp_download(
     studio_path: str,
     local_path: str,
+    recursive: bool = False,
 ) -> None:
     console = Console()
     studio_path_result = parse_studio_path(studio_path)
@@ -146,7 +106,14 @@ def cp_download(
 
     console.print(f"Downloading from {selected_studio.teamspace.name}/{selected_studio.name}")
     if path_info["type"] == "directory":
+        if not recursive:
+            raise ValueError(
+                f"'{studio_path_result['destination']}' is a directory. Use -r flag to copy directories recursively."
+            )
         folder_name = os.path.basename(studio_path_result["destination"].rstrip("/"))
+        if folder_name == "" and local_path in ("./", "."):
+            # handle root directory case (e.g. lit://lightning-ai/gpt-oss/studios/manual-lime-ylu2/)
+            folder_name = selected_studio.name
         target_path = os.path.join(local_path, folder_name)
 
         selected_studio.download_folder(studio_path_result["destination"], target_path)
@@ -161,14 +128,3 @@ def cp_download(
         os.makedirs(os.path.dirname(target_path), exist_ok=True)
         selected_studio.download_file(studio_path_result["destination"], target_path)
         console.print(f"See your file at {target_path}")
-
-
-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/studio/ls.py

@@ -0,0 +1,57 @@
+"""Studio ls command."""
+
+import click
+
+from lightning_sdk.cli.utils.studio_filesystem import parse_studio_path, resolve_studio
+
+
+@click.command("ls")
+@click.argument("path", nargs=1)
+def ls_studio(path: str) -> None:
+    """List contents of a directory in Studio.
+
+    PATH: Studio path in the format
+            lit://<owner>/<teamspace>/studios/<studio>/<directory-path>
+
+    Example:
+        lightning studio ls lit://<owner>/<my-teamspace>/studios/<my-studio>/data
+
+    """
+    return ls_impl(path=path)
+
+
+def ls_impl(path: str) -> None:
+    if not path.startswith("lit://"):
+        raise ValueError("Path must be a Studio path starting with 'lit://'.")
+
+    studio_path_result = parse_studio_path(path)
+    selected_studio = resolve_studio(
+        studio_path_result["studio"], studio_path_result["teamspace"], studio_path_result["owner"]
+    )
+
+    path_info = selected_studio._studio_api.get_path_info(
+        selected_studio._studio.id, selected_studio._teamspace.id, path=studio_path_result["destination"]
+    )
+
+    if not path_info["exists"]:
+        raise FileNotFoundError(
+            f"The provided path does not exist in the studio: {studio_path_result['destination']} "
+            "Note that empty folders may not be detected as existing."
+        )
+
+    if path_info["type"] == "file":
+        # print the file name if it's a file (bash-like behavior)
+        print(studio_path_result["destination"])
+        return
+
+    tree = selected_studio._studio_api.get_tree(
+        selected_studio._studio.id, selected_studio._teamspace.id, path=studio_path_result["destination"]
+    )
+
+    tree_items = tree.get("tree", [])
+
+    for item in tree_items:
+        name = item.get("path", "")
+        if item.get("type") == "tree":
+            name += "/"
+        print(name)

lightning_sdk/cli/studio/rm.py

@@ -0,0 +1,71 @@
+"""Studio rm command."""
+
+import click
+from rich.console import Console
+
+from lightning_sdk.cli.utils.studio_filesystem import parse_studio_path, resolve_studio
+from lightning_sdk.studio import Studio
+
+
+@click.command("rm")
+@click.argument("path", nargs=1)
+@click.option("-r", "--recursive", is_flag=True, help="Remove directories recursively")
+@click.option("-f", "--force", is_flag=True, help="Ignore nonexistent files, never prompt")
+def rm_studio_file(path: str, recursive: bool = False, force: bool = False) -> None:
+    """Remove a Studio file or directory.
+
+    PATH: Studio path to remove. Use the format lit://<owner>/<my-teamspace>/studios/<my-studio>/<filepath>.
+
+    Example:
+        lightning studio rm lit://<owner>/<my-teamspace>/studios/<my-studio>/file.txt
+        lightning studio rm -r lit://<owner>/<my-teamspace>/studios/<my-studio>/folder/
+
+    """
+    return rm_impl(path=path, recursive=recursive, force=force)
+
+
+def rm_impl(path: str, recursive: bool = False, force: bool = False) -> None:
+    if "lit://" not in path:
+        raise ValueError("Path must be a Studio path starting with 'lit://'.")
+
+    console = Console()
+    studio_path_result = parse_studio_path(path)
+
+    selected_studio = resolve_studio(
+        studio_path_result["studio"], studio_path_result["teamspace"], studio_path_result["owner"]
+    )
+
+    # check if file/folder exists
+    path_info = selected_studio._studio_api.get_path_info(
+        selected_studio._studio.id, selected_studio._teamspace.id, path=studio_path_result["destination"]
+    )
+
+    if not path_info["exists"]:
+        if force:
+            # silently ignore nonexistent files with -f flag
+            return
+        raise FileNotFoundError(
+            f"The provided path does not exist in the studio: '{studio_path_result['destination']}'. "
+            "Note that empty folders may not be detected as existing."
+        )
+
+    console.print(f"Removing from {selected_studio.teamspace.name}/{selected_studio.name}")
+
+    if path_info["type"] == "directory":
+        if not recursive:
+            raise ValueError(
+                f"'{studio_path_result['destination']}' is a directory. Use -r flag to remove directories recursively."
+            )
+        rm_folder(selected_studio=selected_studio, path=studio_path_result["destination"], console=console)
+    else:
+        rm_file(selected_studio=selected_studio, path=studio_path_result["destination"], console=console)
+
+
+def rm_file(selected_studio: Studio, path: str, console: Console) -> None:
+    selected_studio._studio_api.remove_file(selected_studio._studio.id, selected_studio._teamspace.id, path)
+    console.print(f"Removed file: {path}")
+
+
+def rm_folder(selected_studio: Studio, path: str, console: Console) -> None:
+    selected_studio._studio_api.remove_folder(selected_studio._studio.id, selected_studio._teamspace.id, path)
+    console.print(f"Removed directory: {path}")

lightning_sdk/cli/utils/logging.py

@@ -30,10 +30,11 @@ def _log_command(message: str = "", duration: int = 0, error: Optional[str] = No
     body = V1CreateSDKCommandHistoryRequest(
         command=original_command,
         duration=duration,
-        message=f"VERSION: {__version__} | {message}",
+        message=f"{message}",
         project_id=None,
         severity=V1SDKCommandHistorySeverity.INFO,
         type=V1SDKCommandHistoryType.CLI,
+        version=__version__,
     )
 
     if error:

lightning_sdk/cli/utils/studio_filesystem.py

@@ -0,0 +1,65 @@
+from typing import Optional, TypedDict
+
+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
+
+
+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:
+        raise ValueError("Invalid: Missing studio name.")
+
+    if len(path_parts) == 1:
+        raise ValueError(
+            "Invalid: Invalid studio path. To refer to the studio root, add a trailing '/' (e.g., 'lit://<owner>/<my-teamspace>/studios/<my-studio>/')"
+        )
+
+    result["studio"] = path_parts[0]
+    result["destination"] = "/".join(path_parts[1:])
+
+    return result
+
+
+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/teamspace_selection.py

@@ -51,6 +51,11 @@ class TeamspacesMenu:
         return TerminalMenu(possible_teamspaces, title=title, clear_menu_on_exit=True)
 
     def _get_possible_teamspaces(self, user: User) -> Dict[str, str]:
+        if self._owner is None:
+            from lightning_sdk.cli.utils.owner_selection import OwnerMenu
+
+            menu = OwnerMenu()
+            self._owner = menu()
         user_api = user._user_api
 
         memberships = user_api._get_all_teamspace_memberships(

lightning_sdk/exceptions.py

@@ -1,2 +1,6 @@
 class OutOfCapacityError(RuntimeError):
     """Raised when the requested machine is not available in the selected cloud account."""
+
+
+class NotSupportedError(RuntimeError):
+    """Raised when the requested machine is not supported in the selected cloud account."""

lightning_sdk/job/base.py

@@ -87,7 +87,7 @@ class _BaseJob(ABC, metaclass=TrackCallsABCMeta):
         cloud_account_auth: bool = False,
         artifacts_local: Optional[str] = None,
         artifacts_remote: Optional[str] = None,
-        entrypoint: str = "sh -c",
+        entrypoint: Optional[str] = None,
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
         cluster: Optional[str] = None,  # deprecated in favor of cloud_account
@@ -193,7 +193,7 @@ class _BaseJob(ABC, metaclass=TrackCallsABCMeta):
                     "Other jobs will automatically persist artifacts to the teamspace distributed filesystem."
                 )
 
-            if entrypoint != "sh -c":
+            if entrypoint is not None:
                 raise ValueError("Specifying the entrypoint has no effect for jobs with Studio envs.")
 
         else:
@@ -201,7 +201,7 @@ class _BaseJob(ABC, metaclass=TrackCallsABCMeta):
                 raise RuntimeError(
                     "image and studio are mutually exclusive as both define the environment to run the job in"
                 )
-            if cloud_account is None and in_studio():
+            if cloud_account is None and cloud_provider is None and in_studio():
                 try:
                     with skip_studio_setup():
                         resolve_studio = Studio(teamspace=teamspace, user=user, org=org)
@@ -219,6 +219,20 @@ class _BaseJob(ABC, metaclass=TrackCallsABCMeta):
                     f"<CONNECTION_TYPE>:<CONNECTION_NAME>:<PATH_WITHIN_CONNECTION>, got {artifacts_local}"
                 )
 
+            # command specified, so use the default entrypoint of sh -c
+            if command is not None and entrypoint is None:
+                entrypoint = "sh -c"
+
+            # entrypoint specifically set to empty string, so set to None here to fall back to the image entrypoint
+            elif entrypoint == "":  # noqa: SIM114
+                entrypoint = None
+
+            # entrypoint not specified, but also no command specified, so use the image entrypoint
+            elif entrypoint is None:
+                entrypoint = None
+
+            # all other cases, the entrypoint has been specifically set, so use it as is
+
         inst = cls(name=name, teamspace=teamspace, org=org, user=user, _fetch_job=False)
         return inst._submit(
             machine=machine,
@@ -255,7 +269,7 @@ class _BaseJob(ABC, metaclass=TrackCallsABCMeta):
         cloud_account_auth: bool = False,
         artifacts_local: Optional[str] = None,
         artifacts_remote: Optional[str] = None,
-        entrypoint: str = "sh -c",
+        entrypoint: Optional[str] = None,
         path_mappings: Optional[Dict[str, str]] = None,
         max_runtime: Optional[int] = None,
         reuse_snapshot: bool = True,
@@ -292,7 +306,12 @@ class _BaseJob(ABC, metaclass=TrackCallsABCMeta):
                 within it.
                 Note that the connection needs to be added to the teamspace already in order for it to be found.
                 Only supported for jobs with a docker image compute environment.
-            entrypoint: The entrypoint of your docker container. Defaults to sh -c.
+            entrypoint: The entrypoint of your docker container. Defaults to `sh -c` which
+                just runs the provided command in a standard shell if a command is provided.
+                If no command is provided, it will run the pre-defined entrypoint of the provided image.
+                To use the pre-defined entrypoint of the provided image with a specified command,
+                set this to an empty string.
+                Only applicable when submitting docker jobs.
                 To use the pre-defined entrypoint of the provided image, set this to an empty string.
                 Only applicable when submitting docker jobs.
             max_runtime: the duration (in seconds) for which to allocate the machine.