diff --git a/studio/backend/utils/paths/storage_roots.py b/studio/backend/utils/paths/storage_roots.py index b7a7a2054d..f1a3038871 100644 --- a/studio/backend/utils/paths/storage_roots.py +++ b/studio/backend/utils/paths/storage_roots.py @@ -28,7 +28,9 @@ def _infer_studio_home_from_venv() -> Path | None: return None candidate = prefix.parent shim_name = "unsloth.exe" if os.name == "nt" else "unsloth" - if (candidate / "share" / "studio.conf").is_file() or (candidate / "bin" / shim_name).exists(): + if (candidate / "share" / "studio.conf").is_file() or ( + candidate / "bin" / shim_name + ).exists(): return candidate return None diff --git a/unsloth_cli/commands/studio.py b/unsloth_cli/commands/studio.py index 5aba3cb49d..63986f8250 100644 --- a/unsloth_cli/commands/studio.py +++ b/unsloth_cli/commands/studio.py @@ -34,10 +34,9 @@ def _looks_like_installer_managed_studio_home(candidate: Path) -> bool: Avoids over-matching on a dev venv that happens to be named unsloth_studio. """ shim_name = "unsloth.exe" if platform.system() == "Windows" else "unsloth" - return ( - (candidate / "share" / "studio.conf").is_file() - or (candidate / "bin" / shim_name).exists() - ) + return (candidate / "share" / "studio.conf").is_file() or ( + candidate / "bin" / shim_name + ).exists() def _resolve_studio_home() -> tuple[Path, bool]: @@ -49,7 +48,9 @@ def _resolve_studio_home() -> tuple[Path, bool]: if prefix.name == "unsloth_studio": inferred = prefix.parent legacy = (Path.home() / ".unsloth" / "studio").resolve() - if inferred != legacy and _looks_like_installer_managed_studio_home(inferred): + if inferred != legacy and _looks_like_installer_managed_studio_home( + inferred + ): return inferred, True except (OSError, ValueError): pass