From 825cbf5f2126657f42603d4de06b3a6d525a8997 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 19 May 2026 10:55:05 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- studio/backend/core/training/worker.py | 4 +++- studio/backend/main.py | 5 ++++- studio/install_python_stack.py | 11 +++++------ 3 files changed, 12 insertions(+), 8 deletions(-) diff --git a/studio/backend/core/training/worker.py b/studio/backend/core/training/worker.py index 3bcf605323..9a0de1cd68 100644 --- a/studio/backend/core/training/worker.py +++ b/studio/backend/core/training/worker.py @@ -104,7 +104,9 @@ if sys.platform == "win32": try: if os.path.isdir(_default_root): - for _ver in sorted(os.listdir(_default_root), key = _ver_key, reverse = True): + for _ver in sorted( + os.listdir(_default_root), key = _ver_key, reverse = True + ): _bin = os.path.join(_default_root, _ver, "bin") if os.path.isdir(_bin): _candidates.append(_bin) diff --git a/studio/backend/main.py b/studio/backend/main.py index 576e15e21e..0e2175200e 100644 --- a/studio/backend/main.py +++ b/studio/backend/main.py @@ -32,6 +32,7 @@ if sys.platform == "win32": _default_root = os.path.join( os.environ.get("ProgramFiles", r"C:\Program Files"), "AMD", "ROCm" ) + def _ver_key(name: str) -> tuple: # Numeric tuple key so "10.0" sorts after "7.0"; non-numeric chunks fall back to string. parts = [] @@ -44,7 +45,9 @@ if sys.platform == "win32": try: if os.path.isdir(_default_root): - for _ver in sorted(os.listdir(_default_root), key = _ver_key, reverse = True): + for _ver in sorted( + os.listdir(_default_root), key = _ver_key, reverse = True + ): _bin = os.path.join(_default_root, _ver, "bin") if os.path.isdir(_bin): candidates.append(_bin) diff --git a/studio/install_python_stack.py b/studio/install_python_stack.py index 43c972b5ba..b9a5031374 100644 --- a/studio/install_python_stack.py +++ b/studio/install_python_stack.py @@ -299,9 +299,10 @@ def _detect_windows_gfx_arch() -> str | None: text = result.stdout.decode(errors = "replace") # findall picks every gcnArchName line so multi-GPU hosts # are enumerable and HIP_VISIBLE_DEVICES selects correctly. - _tokens = [t.strip().lower() for t in re.findall( - r"(?im)^\s*gcnArchName\s*:\s*(\S+)", text - )] + _tokens = [ + t.strip().lower() + for t in re.findall(r"(?im)^\s*gcnArchName\s*:\s*(\S+)", text) + ] _pick = _dedup_pick(_tokens) if _pick: return _pick @@ -678,9 +679,7 @@ def _ensure_rocm_torch() -> None: # first listed GPU. Skip the override unless the resolved GPU is # Strix. _runtime_gfx = ( - gfx_codes[_pick_visible_index(len(gfx_codes))] - if gfx_codes - else None + gfx_codes[_pick_visible_index(len(gfx_codes))] if gfx_codes else None ) if _runtime_gfx in _strix_gfx: _selected_gfx = _runtime_gfx