From 002ea9936b86e1386c9bae495753fdb4b6c7de1a Mon Sep 17 00:00:00 2001 From: Chito Kim Date: Fri, 24 Jul 2026 03:27:08 +0900 Subject: [PATCH 1/7] Make a simple-term-menu wrapper for windows --- .../cli/utils/owner_selection.py | 2 +- .../cli/utils/studio_selection.py | 2 +- .../cli/utils/teamspace_selection.py | 2 +- .../cli/utils/terminal_menu_wrapper.py | 27 +++++++++++++++++++ 4 files changed, 30 insertions(+), 3 deletions(-) create mode 100644 python/lightning_sdk/cli/utils/terminal_menu_wrapper.py diff --git a/python/lightning_sdk/cli/utils/owner_selection.py b/python/lightning_sdk/cli/utils/owner_selection.py index 3bb92805..9027e158 100644 --- a/python/lightning_sdk/cli/utils/owner_selection.py +++ b/python/lightning_sdk/cli/utils/owner_selection.py @@ -3,7 +3,7 @@ from typing import Dict, List, Optional, TypedDict import rich_click as click -from simple_term_menu import TerminalMenu +from .terminal_menu_wrapper import TerminalMenu from lightning_sdk.cli.legacy.exceptions import StudioCliError from lightning_sdk.organization import Organization diff --git a/python/lightning_sdk/cli/utils/studio_selection.py b/python/lightning_sdk/cli/utils/studio_selection.py index 8efe6192..e7a1c3dd 100644 --- a/python/lightning_sdk/cli/utils/studio_selection.py +++ b/python/lightning_sdk/cli/utils/studio_selection.py @@ -3,7 +3,7 @@ from typing import Dict, List, Optional import rich_click as click -from simple_term_menu import TerminalMenu +from .terminal_menu_wrapper import TerminalMenu from lightning_sdk.cli.legacy.exceptions import StudioCliError from lightning_sdk.studio import Studio diff --git a/python/lightning_sdk/cli/utils/teamspace_selection.py b/python/lightning_sdk/cli/utils/teamspace_selection.py index 5b7b72c3..e398ed6c 100644 --- a/python/lightning_sdk/cli/utils/teamspace_selection.py +++ b/python/lightning_sdk/cli/utils/teamspace_selection.py @@ -3,7 +3,7 @@ from typing import Dict, List, Optional import rich_click as click -from simple_term_menu import TerminalMenu +from .terminal_menu_wrapper import TerminalMenu from lightning_sdk.cli.legacy.exceptions import StudioCliError from lightning_sdk.cli.utils.resolve import resolve_teamspace_owner_name_format diff --git a/python/lightning_sdk/cli/utils/terminal_menu_wrapper.py b/python/lightning_sdk/cli/utils/terminal_menu_wrapper.py new file mode 100644 index 00000000..5f1f3efe --- /dev/null +++ b/python/lightning_sdk/cli/utils/terminal_menu_wrapper.py @@ -0,0 +1,27 @@ +import os +from typing import List, Optional + +if os.name == "nt": + import inquirer # import only for windows + class TerminalMenu: + def __init__(self, options: List[str], title: Optional[str] = None, **kwargs): + self.options = options + self.title = title or "Please select an option:" + self.chosen_menu_index: Optional[int] = None + def show(self): + # generate inquirer questons + questions = [ + inquirer.List('choice', + message=self.title, + choices=self.options) + ] + answer = inquirer.prompt(questions) + + # Keyboard Interrupt + if not answer or 'choice' not in answer: + self.chosen_menu_index = None + return + + self.chosen_menu_index = self.options.index(answer['choice']) +else: + from simple_term_menu import TerminalMenu \ No newline at end of file From 4f1d35071f0a4a3081f38d39dc4661bdf8b2cf9c Mon Sep 17 00:00:00 2001 From: Chito Kim Date: Fri, 24 Jul 2026 03:45:10 +0900 Subject: [PATCH 2/7] import terminal menu wrapper to legacy modules --- python/lightning_sdk/cli/legacy/clusters_menu.py | 2 +- python/lightning_sdk/cli/legacy/jobs_menu.py | 2 +- python/lightning_sdk/cli/legacy/mmts_menu.py | 2 +- python/lightning_sdk/cli/legacy/studios_menu.py | 2 +- python/lightning_sdk/cli/legacy/teamspace_menu.py | 2 +- python/lightning_sdk/cli/legacy/upload.py | 2 +- 6 files changed, 6 insertions(+), 6 deletions(-) diff --git a/python/lightning_sdk/cli/legacy/clusters_menu.py b/python/lightning_sdk/cli/legacy/clusters_menu.py index 12624cdf..c795a7af 100644 --- a/python/lightning_sdk/cli/legacy/clusters_menu.py +++ b/python/lightning_sdk/cli/legacy/clusters_menu.py @@ -2,7 +2,7 @@ from typing import List, Optional from rich.console import Console -from simple_term_menu import TerminalMenu +from ..utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk import Teamspace from lightning_sdk.api.cloud_account_api import CloudAccountApi diff --git a/python/lightning_sdk/cli/legacy/jobs_menu.py b/python/lightning_sdk/cli/legacy/jobs_menu.py index 86b4c08e..345c36e1 100644 --- a/python/lightning_sdk/cli/legacy/jobs_menu.py +++ b/python/lightning_sdk/cli/legacy/jobs_menu.py @@ -1,7 +1,7 @@ from typing import Dict, List, Optional from rich.console import Console -from simple_term_menu import TerminalMenu +from ..utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.cli.legacy.exceptions import StudioCliError from lightning_sdk.job import Job diff --git a/python/lightning_sdk/cli/legacy/mmts_menu.py b/python/lightning_sdk/cli/legacy/mmts_menu.py index 042c8ce3..39901b8a 100644 --- a/python/lightning_sdk/cli/legacy/mmts_menu.py +++ b/python/lightning_sdk/cli/legacy/mmts_menu.py @@ -1,7 +1,7 @@ from typing import Dict, List, Optional from rich.console import Console -from simple_term_menu import TerminalMenu +from ..utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.cli.legacy.exceptions import StudioCliError from lightning_sdk.mmt import MMT diff --git a/python/lightning_sdk/cli/legacy/studios_menu.py b/python/lightning_sdk/cli/legacy/studios_menu.py index 609c728e..b065f1b6 100644 --- a/python/lightning_sdk/cli/legacy/studios_menu.py +++ b/python/lightning_sdk/cli/legacy/studios_menu.py @@ -1,7 +1,7 @@ from typing import Dict, List, Optional from rich.console import Console -from simple_term_menu import TerminalMenu +from ..utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk import Studio from lightning_sdk.api import OrgApi, TeamspaceApi diff --git a/python/lightning_sdk/cli/legacy/teamspace_menu.py b/python/lightning_sdk/cli/legacy/teamspace_menu.py index 8bc18384..7b546d41 100644 --- a/python/lightning_sdk/cli/legacy/teamspace_menu.py +++ b/python/lightning_sdk/cli/legacy/teamspace_menu.py @@ -1,7 +1,7 @@ from typing import Dict, List, Optional from rich.console import Console -from simple_term_menu import TerminalMenu +from ..utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.api import OrgApi from lightning_sdk.cli.legacy.exceptions import StudioCliError diff --git a/python/lightning_sdk/cli/legacy/upload.py b/python/lightning_sdk/cli/legacy/upload.py index 58d67483..4414beaa 100644 --- a/python/lightning_sdk/cli/legacy/upload.py +++ b/python/lightning_sdk/cli/legacy/upload.py @@ -9,7 +9,7 @@ import rich from rich.console import Console from rich.progress import Progress, SpinnerColumn, TextColumn, TimeElapsedColumn -from simple_term_menu import TerminalMenu +from ..utils.terminal_menu_wrapper import TerminalMenu from tqdm import tqdm from lightning_sdk.api.lit_container_api import DockerNotRunningError, LCRAuthFailedError, LitContainerApi From 61d4a5d611255f967c20c6fa5e5ae371781709e3 Mon Sep 17 00:00:00 2001 From: Chito Kim Date: Fri, 24 Jul 2026 04:31:15 +0900 Subject: [PATCH 3/7] specify inquirer on dependencies --- python/pyproject.toml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/pyproject.toml b/python/pyproject.toml index 07099853..290a5135 100644 --- a/python/pyproject.toml +++ b/python/pyproject.toml @@ -25,7 +25,8 @@ dependencies = [ # alphabetically sorted "pyyaml", # for config handling "requests", # for downloading files via urls "rich", # for nice printing - "simple-term-menu", # for CLI + "simple-term-menu; sys_platform != 'win32'", # for CLI on POSIX + "inquirer; sys_platform == 'win32'", # for CLI on Windows "six", # for lightning_cloud "tqdm", # for progressbars "urllib3<=2.5.0", # for downloading files via urls From 94d01ec700007ba590b49fb0133474e32a0805fe Mon Sep 17 00:00:00 2001 From: Chito Kim Date: Fri, 24 Jul 2026 04:33:28 +0900 Subject: [PATCH 4/7] code formatting --- .../cli/utils/terminal_menu_wrapper.py | 57 ++++++++++--------- 1 file changed, 30 insertions(+), 27 deletions(-) diff --git a/python/lightning_sdk/cli/utils/terminal_menu_wrapper.py b/python/lightning_sdk/cli/utils/terminal_menu_wrapper.py index 5f1f3efe..726b5f15 100644 --- a/python/lightning_sdk/cli/utils/terminal_menu_wrapper.py +++ b/python/lightning_sdk/cli/utils/terminal_menu_wrapper.py @@ -1,27 +1,30 @@ -import os -from typing import List, Optional - -if os.name == "nt": - import inquirer # import only for windows - class TerminalMenu: - def __init__(self, options: List[str], title: Optional[str] = None, **kwargs): - self.options = options - self.title = title or "Please select an option:" - self.chosen_menu_index: Optional[int] = None - def show(self): - # generate inquirer questons - questions = [ - inquirer.List('choice', - message=self.title, - choices=self.options) - ] - answer = inquirer.prompt(questions) - - # Keyboard Interrupt - if not answer or 'choice' not in answer: - self.chosen_menu_index = None - return - - self.chosen_menu_index = self.options.index(answer['choice']) -else: - from simple_term_menu import TerminalMenu \ No newline at end of file +import os +from typing import Any, List, Optional + +if os.name == "nt": + import inquirer # import only for windows + + class TerminalMenu: + def __init__( + self, + options: List[str], + title: Optional[str] = None, + **kwargs: Any, + ) -> None: + self.options = options + self.title = title or "Please select an option:" + self.chosen_menu_index: Optional[int] = None + + def show(self) -> None: + # generate inquirer questons + questions = [inquirer.List("choice", message=self.title, choices=self.options)] + answer = inquirer.prompt(questions) + + # Keyboard Interrupt + if not answer or "choice" not in answer: + self.chosen_menu_index = None + return + + self.chosen_menu_index = self.options.index(answer["choice"]) +else: + from simple_term_menu import TerminalMenu # noqa: F401 From 1b1b3ba428e2d6787b2a602aa7d5ba7cc4de9521 Mon Sep 17 00:00:00 2001 From: Chito Kim Date: Fri, 24 Jul 2026 05:02:29 +0900 Subject: [PATCH 5/7] switch to absolute imports of terminal menu wrapper --- python/lightning_sdk/cli/legacy/clusters_menu.py | 2 +- python/lightning_sdk/cli/legacy/jobs_menu.py | 2 +- python/lightning_sdk/cli/legacy/mmts_menu.py | 2 +- python/lightning_sdk/cli/legacy/studios_menu.py | 2 +- python/lightning_sdk/cli/legacy/teamspace_menu.py | 2 +- python/lightning_sdk/cli/legacy/upload.py | 2 +- 6 files changed, 6 insertions(+), 6 deletions(-) diff --git a/python/lightning_sdk/cli/legacy/clusters_menu.py b/python/lightning_sdk/cli/legacy/clusters_menu.py index c795a7af..ca3e5cc8 100644 --- a/python/lightning_sdk/cli/legacy/clusters_menu.py +++ b/python/lightning_sdk/cli/legacy/clusters_menu.py @@ -2,7 +2,7 @@ from typing import List, Optional from rich.console import Console -from ..utils.terminal_menu_wrapper import TerminalMenu +from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk import Teamspace from lightning_sdk.api.cloud_account_api import CloudAccountApi diff --git a/python/lightning_sdk/cli/legacy/jobs_menu.py b/python/lightning_sdk/cli/legacy/jobs_menu.py index 345c36e1..1eaac98d 100644 --- a/python/lightning_sdk/cli/legacy/jobs_menu.py +++ b/python/lightning_sdk/cli/legacy/jobs_menu.py @@ -1,7 +1,7 @@ from typing import Dict, List, Optional from rich.console import Console -from ..utils.terminal_menu_wrapper import TerminalMenu +from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.cli.legacy.exceptions import StudioCliError from lightning_sdk.job import Job diff --git a/python/lightning_sdk/cli/legacy/mmts_menu.py b/python/lightning_sdk/cli/legacy/mmts_menu.py index 39901b8a..945b0bab 100644 --- a/python/lightning_sdk/cli/legacy/mmts_menu.py +++ b/python/lightning_sdk/cli/legacy/mmts_menu.py @@ -1,7 +1,7 @@ from typing import Dict, List, Optional from rich.console import Console -from ..utils.terminal_menu_wrapper import TerminalMenu +from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.cli.legacy.exceptions import StudioCliError from lightning_sdk.mmt import MMT diff --git a/python/lightning_sdk/cli/legacy/studios_menu.py b/python/lightning_sdk/cli/legacy/studios_menu.py index b065f1b6..610fa4a9 100644 --- a/python/lightning_sdk/cli/legacy/studios_menu.py +++ b/python/lightning_sdk/cli/legacy/studios_menu.py @@ -1,7 +1,7 @@ from typing import Dict, List, Optional from rich.console import Console -from ..utils.terminal_menu_wrapper import TerminalMenu +from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk import Studio from lightning_sdk.api import OrgApi, TeamspaceApi diff --git a/python/lightning_sdk/cli/legacy/teamspace_menu.py b/python/lightning_sdk/cli/legacy/teamspace_menu.py index 7b546d41..82f2edd6 100644 --- a/python/lightning_sdk/cli/legacy/teamspace_menu.py +++ b/python/lightning_sdk/cli/legacy/teamspace_menu.py @@ -1,7 +1,7 @@ from typing import Dict, List, Optional from rich.console import Console -from ..utils.terminal_menu_wrapper import TerminalMenu +from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.api import OrgApi from lightning_sdk.cli.legacy.exceptions import StudioCliError diff --git a/python/lightning_sdk/cli/legacy/upload.py b/python/lightning_sdk/cli/legacy/upload.py index 4414beaa..647ffb4d 100644 --- a/python/lightning_sdk/cli/legacy/upload.py +++ b/python/lightning_sdk/cli/legacy/upload.py @@ -9,7 +9,7 @@ import rich from rich.console import Console from rich.progress import Progress, SpinnerColumn, TextColumn, TimeElapsedColumn -from ..utils.terminal_menu_wrapper import TerminalMenu +from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from tqdm import tqdm from lightning_sdk.api.lit_container_api import DockerNotRunningError, LCRAuthFailedError, LitContainerApi From 191585066ad5a601a395da2de6a05a669d27c505 Mon Sep 17 00:00:00 2001 From: Chito Kim Date: Fri, 24 Jul 2026 05:14:13 +0900 Subject: [PATCH 6/7] remove relative wrapper imports in the utils as well --- python/lightning_sdk/cli/utils/owner_selection.py | 2 +- python/lightning_sdk/cli/utils/studio_selection.py | 2 +- python/lightning_sdk/cli/utils/teamspace_selection.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/python/lightning_sdk/cli/utils/owner_selection.py b/python/lightning_sdk/cli/utils/owner_selection.py index 9027e158..ae3a7fc0 100644 --- a/python/lightning_sdk/cli/utils/owner_selection.py +++ b/python/lightning_sdk/cli/utils/owner_selection.py @@ -3,7 +3,7 @@ from typing import Dict, List, Optional, TypedDict import rich_click as click -from .terminal_menu_wrapper import TerminalMenu +from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.cli.legacy.exceptions import StudioCliError from lightning_sdk.organization import Organization diff --git a/python/lightning_sdk/cli/utils/studio_selection.py b/python/lightning_sdk/cli/utils/studio_selection.py index e7a1c3dd..9a82c7af 100644 --- a/python/lightning_sdk/cli/utils/studio_selection.py +++ b/python/lightning_sdk/cli/utils/studio_selection.py @@ -3,7 +3,7 @@ from typing import Dict, List, Optional import rich_click as click -from .terminal_menu_wrapper import TerminalMenu +from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.cli.legacy.exceptions import StudioCliError from lightning_sdk.studio import Studio diff --git a/python/lightning_sdk/cli/utils/teamspace_selection.py b/python/lightning_sdk/cli/utils/teamspace_selection.py index e398ed6c..f6d5cf1b 100644 --- a/python/lightning_sdk/cli/utils/teamspace_selection.py +++ b/python/lightning_sdk/cli/utils/teamspace_selection.py @@ -3,7 +3,7 @@ from typing import Dict, List, Optional import rich_click as click -from .terminal_menu_wrapper import TerminalMenu +from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.cli.legacy.exceptions import StudioCliError from lightning_sdk.cli.utils.resolve import resolve_teamspace_owner_name_format From 62315e7eb7969b89c9742c10647fc107c0c68ddd Mon Sep 17 00:00:00 2001 From: Chito Kim Date: Fri, 24 Jul 2026 05:21:41 +0900 Subject: [PATCH 7/7] code foramtting again --- python/lightning_sdk/cli/legacy/clusters_menu.py | 2 +- python/lightning_sdk/cli/legacy/jobs_menu.py | 2 +- python/lightning_sdk/cli/legacy/mmts_menu.py | 2 +- python/lightning_sdk/cli/legacy/studios_menu.py | 2 +- python/lightning_sdk/cli/legacy/teamspace_menu.py | 2 +- python/lightning_sdk/cli/legacy/upload.py | 2 +- python/lightning_sdk/cli/utils/owner_selection.py | 2 +- python/lightning_sdk/cli/utils/studio_selection.py | 2 +- python/lightning_sdk/cli/utils/teamspace_selection.py | 2 +- 9 files changed, 9 insertions(+), 9 deletions(-) diff --git a/python/lightning_sdk/cli/legacy/clusters_menu.py b/python/lightning_sdk/cli/legacy/clusters_menu.py index ca3e5cc8..20ca3c07 100644 --- a/python/lightning_sdk/cli/legacy/clusters_menu.py +++ b/python/lightning_sdk/cli/legacy/clusters_menu.py @@ -2,10 +2,10 @@ from typing import List, Optional from rich.console import Console -from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk import Teamspace from lightning_sdk.api.cloud_account_api import CloudAccountApi +from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.lightning_cloud.openapi import V1ClusterType, V1ProjectClusterBinding diff --git a/python/lightning_sdk/cli/legacy/jobs_menu.py b/python/lightning_sdk/cli/legacy/jobs_menu.py index 1eaac98d..eaa19ca8 100644 --- a/python/lightning_sdk/cli/legacy/jobs_menu.py +++ b/python/lightning_sdk/cli/legacy/jobs_menu.py @@ -1,9 +1,9 @@ from typing import Dict, List, Optional from rich.console import Console -from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.cli.legacy.exceptions import StudioCliError +from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.job import Job from lightning_sdk.teamspace import Teamspace diff --git a/python/lightning_sdk/cli/legacy/mmts_menu.py b/python/lightning_sdk/cli/legacy/mmts_menu.py index 945b0bab..057808ff 100644 --- a/python/lightning_sdk/cli/legacy/mmts_menu.py +++ b/python/lightning_sdk/cli/legacy/mmts_menu.py @@ -1,9 +1,9 @@ from typing import Dict, List, Optional from rich.console import Console -from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.cli.legacy.exceptions import StudioCliError +from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.mmt import MMT from lightning_sdk.teamspace import Teamspace diff --git a/python/lightning_sdk/cli/legacy/studios_menu.py b/python/lightning_sdk/cli/legacy/studios_menu.py index 610fa4a9..0d55d378 100644 --- a/python/lightning_sdk/cli/legacy/studios_menu.py +++ b/python/lightning_sdk/cli/legacy/studios_menu.py @@ -1,10 +1,10 @@ from typing import Dict, List, Optional from rich.console import Console -from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk import Studio from lightning_sdk.api import OrgApi, TeamspaceApi +from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.user import User diff --git a/python/lightning_sdk/cli/legacy/teamspace_menu.py b/python/lightning_sdk/cli/legacy/teamspace_menu.py index 82f2edd6..81df2847 100644 --- a/python/lightning_sdk/cli/legacy/teamspace_menu.py +++ b/python/lightning_sdk/cli/legacy/teamspace_menu.py @@ -1,10 +1,10 @@ from typing import Dict, List, Optional from rich.console import Console -from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.api import OrgApi from lightning_sdk.cli.legacy.exceptions import StudioCliError +from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.teamspace import Teamspace from lightning_sdk.user import User from lightning_sdk.utils.resolve import _get_authed_user diff --git a/python/lightning_sdk/cli/legacy/upload.py b/python/lightning_sdk/cli/legacy/upload.py index 647ffb4d..8731a77c 100644 --- a/python/lightning_sdk/cli/legacy/upload.py +++ b/python/lightning_sdk/cli/legacy/upload.py @@ -9,7 +9,6 @@ import rich from rich.console import Console from rich.progress import Progress, SpinnerColumn, TextColumn, TimeElapsedColumn -from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from tqdm import tqdm from lightning_sdk.api.lit_container_api import DockerNotRunningError, LCRAuthFailedError, LitContainerApi @@ -17,6 +16,7 @@ from lightning_sdk.cli.legacy.exceptions import StudioCliError from lightning_sdk.cli.legacy.studios_menu import _StudiosMenu from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu +from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.constants import _LIGHTNING_DEBUG from lightning_sdk.exceptions import DeprecatedCommand, DeprecatedError from lightning_sdk.models import upload_model as _upload_model diff --git a/python/lightning_sdk/cli/utils/owner_selection.py b/python/lightning_sdk/cli/utils/owner_selection.py index ae3a7fc0..631d9819 100644 --- a/python/lightning_sdk/cli/utils/owner_selection.py +++ b/python/lightning_sdk/cli/utils/owner_selection.py @@ -3,9 +3,9 @@ from typing import Dict, List, Optional, TypedDict import rich_click as click -from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.cli.legacy.exceptions import StudioCliError +from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.organization import Organization from lightning_sdk.owner import Owner from lightning_sdk.user import User diff --git a/python/lightning_sdk/cli/utils/studio_selection.py b/python/lightning_sdk/cli/utils/studio_selection.py index 9a82c7af..9635a0d9 100644 --- a/python/lightning_sdk/cli/utils/studio_selection.py +++ b/python/lightning_sdk/cli/utils/studio_selection.py @@ -3,9 +3,9 @@ from typing import Dict, List, Optional import rich_click as click -from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.cli.legacy.exceptions import StudioCliError +from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.studio import Studio from lightning_sdk.teamspace import Teamspace from lightning_sdk.utils.resolve import _get_authed_user diff --git a/python/lightning_sdk/cli/utils/teamspace_selection.py b/python/lightning_sdk/cli/utils/teamspace_selection.py index f6d5cf1b..870e9c25 100644 --- a/python/lightning_sdk/cli/utils/teamspace_selection.py +++ b/python/lightning_sdk/cli/utils/teamspace_selection.py @@ -3,10 +3,10 @@ from typing import Dict, List, Optional import rich_click as click -from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.cli.legacy.exceptions import StudioCliError from lightning_sdk.cli.utils.resolve import resolve_teamspace_owner_name_format +from lightning_sdk.cli.utils.terminal_menu_wrapper import TerminalMenu from lightning_sdk.teamspace import Organization, Teamspace from lightning_sdk.user import Owner, User from lightning_sdk.utils.resolve import ApiException, _get_authed_user, _resolve_teamspace