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

lightning_sdk/cli/inspect.py

@@ -1,9 +1,12 @@
 from typing import Optional
 
+import click
+from rich.console import Console
+
 from lightning_sdk.cli.job_and_mmt_action import _JobAndMMTAction
 
 
-class _Inspect(_JobAndMMTAction):
+class _Inspect:
     """Inspect resources of the Lightning AI platform to get additional details as JSON."""
 
     def job(self, name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
@@ -16,7 +19,7 @@ class _Inspect(_JobAndMMTAction):
                 If not specified can be selected interactively.
 
         """
-        print(super().job(name=name, teamspace=teamspace).json())
+        job(name=name, teamspace=teamspace)
 
     def mmt(self, name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
         """Inspect a multi-machine job for further details as JSON.
@@ -28,4 +31,43 @@ class _Inspect(_JobAndMMTAction):
                 If not specified can be selected interactively.
 
         """
-        print(super().mmt(name=name, teamspace=teamspace).json())
+        mmt(name=name, teamspace=teamspace)
+
+
+@click.group(name="inspect")
+def inspect() -> None:
+    """Inspect resources of the Lightning AI platform to get additional details as JSON."""
+
+
+# @inspect.command(name="job")
+# @click.option("--name", default=None, help="the name of the job. If not specified can be selected interactively.")
+# @click.option(
+#     "--teamspace",
+#     default=None,
+#     help=(
+#         "the name of the teamspace the job lives in."
+#         "Should be specified as {teamspace_owner}/{teamspace_name} (e.g my-org/my-teamspace). "
+#         "If not specified can be selected interactively."
+#     ),
+# )
+def job(name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
+    """Inspect a job for further details as JSON."""
+    menu = _JobAndMMTAction()
+    Console().print(menu.job(name=name, teamspace=teamspace).json())
+
+
+# @inspect.command(name="mmt")
+# @click.option("--name", default=None, help="the name of the job. If not specified can be selected interactively.")
+# @click.option(
+#     "--teamspace",
+#     default=None,
+#     help=(
+#         "the name of the teamspace the job lives in."
+#         "Should be specified as {teamspace_owner}/{teamspace_name} (e.g my-org/my-teamspace). "
+#         "If not specified can be selected interactively."
+#     ),
+# )
+def mmt(name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
+    """Inspect a multi-machine job for further details as JSON."""
+    menu = _JobAndMMTAction()
+    Console().print(menu.mmt(name=name, teamspace=teamspace).json())

lightning_sdk/cli/jobs_menu.py

@@ -1,5 +1,6 @@
 from typing import Dict, List, Optional
 
+from rich.console import Console
 from simple_term_menu import TerminalMenu
 
 from lightning_sdk.cli.exceptions import StudioCliError
@@ -20,7 +21,7 @@ class _JobsMenu:
             if j.name == job:
                 return j
 
-        print("Could not find Job {job}, please select it from the list:")
+        Console().print("Could not find Job {job}, please select it from the list:")
         return self._get_job_from_interactive_menu(possible_jobs)
 
     @staticmethod

lightning_sdk/cli/list.py

@@ -1,27 +1,58 @@
-from typing import Optional
+from contextlib import suppress
+from typing import Callable, Optional
 
 from rich.console import Console
 from rich.table import Table
+from typing_extensions import Literal
 
-from lightning_sdk import Machine
+from lightning_sdk import Job, Machine, Studio, Teamspace
 from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
+from lightning_sdk.lightning_cloud.openapi import V1MultiMachineJob
 from lightning_sdk.lit_container import LitContainer
+from lightning_sdk.utils.resolve import _get_authed_user
 
 
 class _List(_TeamspacesMenu):
     """List resources on the Lightning AI platform."""
 
-    def studios(self, teamspace: Optional[str] = None) -> None:
+    @staticmethod
+    def _sort_studios_key(sort_by: str) -> Callable[[Studio], str]:
+        """Return a key function to sort studios by a given attribute."""
+        sort_key_map = {
+            "name": lambda s: str(s.name or ""),
+            "teamspace": lambda s: str(s.teamspace.name or ""),
+            "status": lambda s: str(s.status or ""),
+            "machine": lambda s: str(s.machine or ""),
+            "cloud-account": lambda s: str(s.cloud_account or ""),
+        }
+        return sort_key_map.get(sort_by, lambda s: s.name)
+
+    def studios(
+        self,
+        teamspace: Optional[str] = None,
+        all: bool = False,  # noqa: A002
+        sort_by: Optional[Literal["name", "teamspace", "status", "machine", "cloud-account"]] = None,
+    ) -> None:
         """List studios for a given teamspace.
 
         Args:
             teamspace: the teamspace to list studios from. Should be specified as {owner}/{name}
                 If not provided, can be selected in an interactive menu.
+            all: if teamspece is not provided, list all studios in all teamspaces.
+            sort_by: the attribute to sort the studios by.
+                Can be one of "name", "teamspace", "status", "machine", "cloud-account".
 
         """
-        resolved_teamspace = self._resolve_teamspace(teamspace=teamspace)
-
-        studios = resolved_teamspace.studios
+        studios = []
+        if all and not teamspace:
+            user = _get_authed_user()
+            possible_teamspaces = self._get_possible_teamspaces(user)
+            for ts in possible_teamspaces.values():
+                teamspace = Teamspace(**ts)
+                studios.extend(teamspace.studios)
+        else:
+            resolved_teamspace = self._resolve_teamspace(teamspace=teamspace)
+            studios = resolved_teamspace.studios
 
         table = Table(
             pad_edge=True,
@@ -31,33 +62,59 @@ class _List(_TeamspacesMenu):
         table.add_column("Status")
         table.add_column("Machine")
         table.add_column("Cloud account")
-        for studio in studios:
+        for studio in sorted(studios, key=self._sort_studios_key(sort_by)):
             table.add_row(
                 studio.name,
                 f"{studio.teamspace.owner.name}/{studio.teamspace.name}",
                 str(studio.status),
-                str(studio.machine) if studio.machine is not None else None,
+                str(studio.machine) if studio.machine is not None else None,  # when None the cell is empty
                 str(studio.cloud_account),
             )
 
-        console = Console()
-        console.print(table)
-
-    def jobs(self, teamspace: Optional[str] = None) -> None:
+        Console().print(table)
+
+    @staticmethod
+    def _sort_jobs_key(sort_by: str) -> Callable[[Job], str]:
+        """Return a key function to sort studios by a given attribute."""
+        sort_key_map = {
+            "name": lambda j: str(j.name or ""),
+            "teamspace": lambda j: str(j.teamspace.name or ""),
+            "status": lambda j: str(j.status or ""),
+            "machine": lambda j: str(j.machine or ""),
+            "studio": lambda j: str(j.studio or ""),
+            "image": lambda j: str(j.image or ""),
+            "cloud-account": lambda j: str(j.cloud_account or ""),
+        }
+        return sort_key_map.get(sort_by, lambda j: j.name)
+
+    def jobs(
+        self,
+        teamspace: Optional[str] = None,
+        all: bool = False,  # noqa: A002
+        sort_by: Optional[Literal["name", "teamspace", "status", "studio", "machine", "image", "cloud-account"]] = None,
+    ) -> None:
         """List jobs for a given teamspace.
 
         Args:
             teamspace: the teamspace to list jobs from. Should be specified as {owner}/{name}
                 If not provided, can be selected in an interactive menu.
+            all: if teamspece is not provided, list all jobs in all teamspaces.
+            sort_by: the attribute to sort the jobs by.
+                Can be one of "name", "teamspace", "status", "studio", "machine", "image", "cloud-account".
 
         """
-        resolved_teamspace = self._resolve_teamspace(teamspace=teamspace)
-
-        jobs = resolved_teamspace.jobs
+        jobs = []
+        if all and not teamspace:
+            user = _get_authed_user()
+            possible_teamspaces = self._get_possible_teamspaces(user)
+            for ts in possible_teamspaces.values():
+                teamspace = Teamspace(**ts)
+                jobs.extend(teamspace.jobs)
+        else:
+            resolved_teamspace = self._resolve_teamspace(teamspace=teamspace)
+            jobs = resolved_teamspace.jobs
 
-        table = Table(
-            pad_edge=True,
-        )
+        table = Table(pad_edge=True)
         table.add_column("Name")
         table.add_column("Teamspace")
         table.add_column("Studio")
@@ -65,36 +122,64 @@ class _List(_TeamspacesMenu):
         table.add_column("Status")
         table.add_column("Machine")
         table.add_column("Total Cost")
-        for j in jobs:
+        for j in sorted(jobs, key=self._sort_jobs_key(sort_by)):
             # we know we just fetched these, so no need to refetch
             j._prevent_refetch_latest = True
             j._internal_job._prevent_refetch_latest = True
 
             studio = j.studio
-            table.add_row(
-                j.name,
-                f"{j.teamspace.owner.name}/{j.teamspace.name}",
-                studio.name if studio else None,
-                j.image,
-                str(j.status),
-                str(j.machine),
-                f"{j.total_cost:.3f}",
-            )
-
-        console = Console()
-        console.print(table)
-
-    def mmts(self, teamspace: Optional[str] = None) -> None:
+            with suppress(RuntimeError):
+                table.add_row(
+                    j.name,
+                    f"{j.teamspace.owner.name}/{j.teamspace.name}",
+                    studio.name if studio else None,
+                    j.image,
+                    str(j.status) if j.status is not None else None,
+                    str(j.machine),
+                    f"{j.total_cost:.3f}",
+                )
+
+        Console().print(table)
+
+    def _sort_mmts_key(self, sort_by: str) -> Callable[[V1MultiMachineJob], str]:
+        """Return a key function to sort multi-machine jobs by a given attribute."""
+        sort_key_map = {
+            "name": lambda j: str(j.name or ""),
+            "teamspace": lambda j: str(j.teamspace.name or ""),
+            "studio": lambda j: str(j.studio.name or ""),
+            "image": lambda j: str(j.image or ""),
+            "status": lambda j: str(j.status or ""),
+            "machine": lambda j: str(j.machine or ""),
+            "cloud-account": lambda j: str(j.cloud_account or ""),
+        }
+        return sort_key_map.get(sort_by, lambda j: j.name)
+
+    def mmts(
+        self,
+        teamspace: Optional[str] = None,
+        all: bool = False,  # noqa: A002
+        sort_by: Optional[Literal["name", "teamspace", "studio", "image", "status", "machine", "cloud-account"]] = None,
+    ) -> None:
         """List multi-machine jobs for a given teamspace.
 
         Args:
             teamspace: the teamspace to list jobs from. Should be specified as {owner}/{name}
                 If not provided, can be selected in an interactive menu.
+            all: if teamspece is not provided, list all multi-machine jobs in all teamspaces.
+            sort_by: the attribute to sort the multi-machine jobs by.
+                Can be one of "name", "teamspace", "studio", "image", "status", "machine", "cloud-account".
 
         """
-        resolved_teamspace = self._resolve_teamspace(teamspace=teamspace)
-
-        jobs = resolved_teamspace.multi_machine_jobs
+        jobs = []
+        if all and not teamspace:
+            user = _get_authed_user()
+            possible_teamspaces = self._get_possible_teamspaces(user)
+            for ts in possible_teamspaces.values():
+                teamspace = Teamspace(**ts)
+                jobs.extend(teamspace.multi_machine_jobs)
+        else:
+            resolved_teamspace = self._resolve_teamspace(teamspace=teamspace)
+            jobs = resolved_teamspace.multi_machine_jobs
 
         table = Table(pad_edge=True)
         table.add_column("Name")
@@ -105,25 +190,26 @@ class _List(_TeamspacesMenu):
         table.add_column("Machine")
         table.add_column("Num Machines")
         table.add_column("Total Cost")
-        for j in jobs:
+        for j in sorted(jobs, key=self._sort_mmts_key(sort_by)):
             # we know we just fetched these, so no need to refetch
             j._prevent_refetch_latest = True
-            j._internal_job._prevent_refetch_latest = True
+            with suppress(AttributeError):
+                j._internal_job._prevent_refetch_latest = True
 
             studio = j.studio
-            table.add_row(
-                j.name,
-                f"{j.teamspace.owner.name}/{j.teamspace.name}",
-                studio.name if studio else None,
-                j.image,
-                str(j.status),
-                str(j.machine),
-                str(j.num_machines),
-                str(j.total_cost),
-            )
-
-        console = Console()
-        console.print(table)
+            with suppress(RuntimeError):
+                table.add_row(
+                    j.name,
+                    f"{j.teamspace.owner.name}/{j.teamspace.name}",
+                    studio.name if studio else None,
+                    j.image,
+                    str(j.status),
+                    str(j.machine),
+                    str(j.num_machines),
+                    str(j.total_cost),
+                )
+
+        Console().print(table)
 
     def containers(self, teamspace: Optional[str] = None) -> None:
         """Display the list of available containers.
@@ -141,8 +227,7 @@ class _List(_TeamspacesMenu):
         table.add_column("CREATED")
         for repo in result:
             table.add_row(repo["REPOSITORY"], repo["IMAGE ID"], repo["CREATED"])
-        console = Console()
-        console.print(table)
+        Console().print(table)
 
     def machines(self) -> None:
         """Display the list of available machines."""
@@ -156,5 +241,4 @@ class _List(_TeamspacesMenu):
         for name in sorted(machine_types):
             table.add_row(name)
 
-        console = Console()
-        console.print(table)
+        Console().print(table)

lightning_sdk/cli/mmts_menu.py

@@ -1,5 +1,6 @@
 from typing import Dict, List, Optional
 
+from rich.console import Console
 from simple_term_menu import TerminalMenu
 
 from lightning_sdk.cli.exceptions import StudioCliError
@@ -20,7 +21,7 @@ class _MMTsMenu:
             if j.name == mmt:
                 return j
 
-        print("Could not find Multi-Machine Job {mmt}, please select it from the list:")
+        Console().print("Could not find Multi-Machine Job {mmt}, please select it from the list:")
         return self._get_mmt_from_interactive_menu(possible_mmts)
 
     @staticmethod

lightning_sdk/cli/run.py

@@ -8,7 +8,7 @@ from lightning_sdk.teamspace import Teamspace
 if TYPE_CHECKING:
     from lightning_sdk.cli.legacy import _LegacyLightningCLI
 
-_MACHINE_VALUES = tuple([machine.value for machine in Machine])
+_MACHINE_VALUES = tuple([machine.name for machine in Machine.__dict__.values() if isinstance(machine, Machine)])
 
 
 class _Run:
@@ -156,16 +156,12 @@ class _Run:
             machine = "CPU"
         machine_enum: Union[str, Machine]
         try:
-            machine_enum = Machine[machine.upper()]
+            machine_enum = getattr(Machine, machine.upper(), Machine(machine, machine))
         except KeyError:
             machine_enum = machine
 
         resolved_teamspace = Teamspace(name=teamspace, org=org, user=user)
 
-        if cloud_account is None:
-            cloud_account = resolved_teamspace.default_cloud_account
-        machine_enum = Machine(machine.upper())
-
         path_mappings_dict = self._resolve_path_mapping(path_mappings=path_mappings)
 
         Job.run(
@@ -222,13 +218,11 @@ class _Run:
             machine = "CPU"
         machine_enum: Union[str, Machine]
         try:
-            machine_enum = Machine[machine.upper()]
+            machine_enum = getattr(Machine, machine.upper(), Machine(machine, machine))
         except KeyError:
             machine_enum = machine
 
         resolved_teamspace = Teamspace(name=teamspace, org=org, user=user)
-        if cloud_account is None:
-            cloud_account = resolved_teamspace.default_cloud_account
 
         if image is None:
             raise RuntimeError("Image needs to be specified to run a multi-machine job")

lightning_sdk/cli/serve.py

@@ -167,7 +167,6 @@ class _Docker:
         import litserve as ls
         from litserve import docker_builder
 
-        console = Console()
         requirements = ""
         if os.path.exists("requirements.txt"):
             requirements = "-r requirements.txt"
@@ -210,5 +209,5 @@ Update [underline]{os.path.abspath("Dockerfile")}[/underline] to add any additio
 [bold]To push the container to a registry:[/bold]
 > [underline]docker push {tag}[/underline]
 """
-        console.print(success_msg)
+        Console().print(success_msg)
         return os.path.abspath("Dockerfile")

lightning_sdk/cli/start.py

@@ -7,7 +7,7 @@ class _Start:
     """Start resources on the Lightning AI platform."""
 
     def __init__(self) -> None:
-        _machine_values = tuple([machine.value for machine in Machine])
+        _machine_values = tuple([machine.name for machine in Machine.__dict__.values() if isinstance(machine, Machine)])
 
         docstr_studio = f"""Start a studio on a given machine.
 
@@ -36,7 +36,7 @@ class _Start:
             studio = Studio(name=name, teamspace=teamspace, org=None, user=owner, create_ok=False)
 
         try:
-            resolved_machine = Machine[machine.upper()]
+            resolved_machine = getattr(Machine, machine.upper(), Machine(machine, machine))
         except KeyError:
             resolved_machine = machine
 

lightning_sdk/cli/stop.py

@@ -1,5 +1,7 @@
 from typing import Optional
 
+from rich.console import Console
+
 from lightning_sdk.cli.job_and_mmt_action import _JobAndMMTAction
 from lightning_sdk.studio import Studio
 
@@ -20,7 +22,7 @@ class _Stop(_JobAndMMTAction):
         job = super().job(name=name, teamspace=teamspace)
 
         job.stop()
-        print(f"Successfully stopped {job.name}!")
+        Console().print(f"Successfully stopped {job.name}!")
 
     def mmt(self, name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
         """Stop a multi-machine job.
@@ -35,7 +37,7 @@ class _Stop(_JobAndMMTAction):
         mmt = super().mmt(name=name, teamspace=teamspace)
 
         mmt.stop()
-        print(f"Successfully stopped {mmt.name}!")
+        Console().print(f"Successfully stopped {mmt.name}!")
 
     def studio(self, name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
         """Stop a running studio.
@@ -60,4 +62,4 @@ class _Stop(_JobAndMMTAction):
             studio = Studio(name=name, teamspace=teamspace, org=None, user=owner, create_ok=False)
 
         studio.stop()
-        print("Studio successfully stopped")
+        Console().print("Studio successfully stopped")

lightning_sdk/cli/studios_menu.py

@@ -1,7 +1,9 @@
 from typing import Dict, List, Optional
 
+from rich.console import Console
 from simple_term_menu import TerminalMenu
 
+from lightning_sdk import Studio
 from lightning_sdk.api import OrgApi, TeamspaceApi
 from lightning_sdk.user import User
 
@@ -18,7 +20,7 @@ class _StudiosMenu:
             if st["teamspace"] == teamspace and name == st["name"]:
                 return st
 
-        print("Could not find Studio {studio}, please select it from the list:")
+        Console().print("Could not find Studio {studio}, please select it from the list:")
         return self._get_studio_from_interactive_menu(possible_studios)
 
     @staticmethod
@@ -76,3 +78,24 @@ class _StudiosMenu:
                 possible_studios.append({"name": st.name, **teamspaces[teamspace_id]})
 
         return possible_studios
+
+    def _get_studio(self, name: str, teamspace: str) -> Studio:
+        """Get studio object from name and teamspace.
+
+        Args:
+            name: Name of the studio
+            teamspace: Name of the teamspace
+        """
+        if teamspace:
+            ts_splits = teamspace.split("/")
+            if len(ts_splits) != 2:
+                raise ValueError(f"Teamspace should be of format <OWNER>/<TEAMSPACE_NAME> but got {teamspace}")
+            owner, teamspace = ts_splits
+        else:
+            owner, teamspace = None, None
+
+        try:
+            studio = Studio(name=name, teamspace=teamspace, org=owner, user=None, create_ok=False)
+        except (RuntimeError, ValueError):
+            studio = Studio(name=name, teamspace=teamspace, org=None, user=owner, create_ok=False)
+        return studio

lightning_sdk/cli/switch.py

@@ -7,7 +7,7 @@ class _Switch:
     """Switch machines for resources on the Lightning AI platform."""
 
     def __init__(self) -> None:
-        _machine_values = tuple([machine.value for machine in Machine])
+        _machine_values = tuple([machine.name for machine in Machine.__dict__.values() if isinstance(machine, Machine)])
 
         docstr_studio = f"""Switch a studio to a given machine.
 
@@ -36,7 +36,7 @@ class _Switch:
             studio = Studio(name=name, teamspace=teamspace, org=None, user=owner, create_ok=False)
 
         try:
-            resolved_machine = Machine[machine.upper()]
+            resolved_machine = getattr(Machine, machine.upper(), Machine(machine, machine))
         except KeyError:
             resolved_machine = machine
 

lightning_sdk/cli/teamspace_menu.py

@@ -1,5 +1,6 @@
 from typing import Dict, List, Optional
 
+from rich.console import Console
 from simple_term_menu import TerminalMenu
 
 from lightning_sdk.api import OrgApi
@@ -26,7 +27,7 @@ class _TeamspacesMenu:
             if ts["name"] == name and (ts["user"] == owner or ts["org"] == owner):
                 return ts
 
-        print("Could not find Teamspace {teamspace}, please select it from the list:")
+        Console().print(f"Could not find Teamspace {teamspace}, please select it from the list:")
         return self._get_teamspace_from_interactive_menu(possible_teamspaces)
 
     @staticmethod

lightning_sdk/cli/upload.py

@@ -71,6 +71,7 @@ class _Uploads(_StudiosMenu, _TeamspacesMenu):
             If not specified, will use the file or directory name of the path you want to upload
             and place it in your home directory.
         """
+        console = Console()
         if remote_path is None:
             remote_path = os.path.basename(path)
 
@@ -81,7 +82,7 @@ class _Uploads(_StudiosMenu, _TeamspacesMenu):
 
         selected_studio = self._resolve_studio(studio)
 
-        print(f"Uploading to {selected_studio.teamspace.name}/{selected_studio.name}")
+        console.print(f"Uploading to {selected_studio.teamspace.name}/{selected_studio.name}")
 
         pairs = {}
         for root, _, files in os.walk(path):
@@ -112,7 +113,7 @@ class _Uploads(_StudiosMenu, _TeamspacesMenu):
             + "/studios/"
             + selected_studio.name
         )
-        print(f"See your files at {studio_url}")
+        console.print(f"See your files at {studio_url}")
 
     def file(self, path: str, studio: Optional[str] = None, remote_path: Optional[str] = None) -> None:
         """Upload a file to a Studio.
@@ -126,6 +127,7 @@ class _Uploads(_StudiosMenu, _TeamspacesMenu):
             If not specified, will use the name of the file you want to upload
             and place it in your home directory.
         """
+        console = Console()
         if remote_path is None:
             remote_path = os.path.basename(path)
 
@@ -136,7 +138,7 @@ class _Uploads(_StudiosMenu, _TeamspacesMenu):
 
         selected_studio = self._resolve_studio(studio)
 
-        print(f"Uploading to {selected_studio.teamspace.name}/{selected_studio.name}")
+        console.print(f"Uploading to {selected_studio.teamspace.name}/{selected_studio.name}")
 
         self._single_file_upload(selected_studio, path, remote_path, True)
 
@@ -149,7 +151,7 @@ class _Uploads(_StudiosMenu, _TeamspacesMenu):
             + "/studios/"
             + selected_studio.name
         )
-        print(f"See your file at {studio_url}")
+        console.print(f"See your file at {studio_url}")
 
     def container(self, container: str, tag: str = "latest", teamspace: Optional[str] = None) -> None:
         teamspace = self._resolve_teamspace(teamspace)

lightning_sdk/helpers.py

@@ -1,8 +1,13 @@
 import functools
+import importlib
+import os
+import sys
 import warnings
 from typing import Optional
 
 import requests
+import tqdm
+import tqdm.std
 from packaging import version as packaging_version
 
 __package_name__ = "lightning-sdk"
@@ -47,3 +52,18 @@ def _check_version_and_prompt_upgrade(curr_version: str) -> None:
             UserWarning,
         )
     return
+
+
+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()):
+        os.unsetenv("TQDM_POSITION")
+        os.unsetenv("TQDM_MININTERVAL")
+    else:
+        # makes use of https://github.com/tqdm/tqdm/blob/master/tqdm/utils.py#L34 to set defaults
+        os.environ.update({"TQDM_POSITION": "-1", "TQDM_MININTERVAL": "1"})
+
+    # reload to make sure env vars are parsed again
+    importlib.reload(tqdm.std)
+    importlib.reload(tqdm)

lightning_sdk/job/job.py

@@ -235,7 +235,7 @@ class Job(_BaseJob):
         return self._internal_job.delete()
 
     @property
-    def status(self) -> "Status":
+    def status(self) -> Optional["Status"]:
         """The current status of the job."""
         return self._internal_job.status