diff --git a/studio/backend/run.py b/studio/backend/run.py index e906e0ad42..67c689699b 100644 --- a/studio/backend/run.py +++ b/studio/backend/run.py @@ -774,6 +774,7 @@ def _recorded_studio_pids() -> "set[int]": pids.add(int(text)) return pids + # Direct backend launches bypass the CLI's env re-export; do it here for # real custom roots so unsloth-zoo's import-time LLAMA_CPP_DEFAULT_DIR # picks up the custom build. Skip legacy-default to avoid flipping diff --git a/unsloth_cli/tests/test_studio_stop.py b/unsloth_cli/tests/test_studio_stop.py index c926b74aae..26140fc9a7 100644 --- a/unsloth_cli/tests/test_studio_stop.py +++ b/unsloth_cli/tests/test_studio_stop.py @@ -26,7 +26,13 @@ def _studio(): return _studio_mod -def _install(monkeypatch, tmp_path, *, alive, killed = None): +def _install( + monkeypatch, + tmp_path, + *, + alive, + killed = None, +): """Point the CLI at tmp_path and fake process liveness.""" studio_mod = _studio() monkeypatch.setattr(studio_mod, "STUDIO_HOME", tmp_path)