From 14f655917c653b42e96eddc241368e7dfd5a5b13 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Wed, 6 May 2026 21:32:22 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 studio/install_python_stack.py | 19 +++++++++++++++----
 1 file changed, 15 insertions(+), 4 deletions(-)

diff --git a/studio/install_python_stack.py b/studio/install_python_stack.py
index 0c6960adf2..0be4607380 100644
--- a/studio/install_python_stack.py
+++ b/studio/install_python_stack.py
@@ -275,7 +275,11 @@ def _ensure_rocm_torch() -> None:
             return
         try:
             probe = subprocess.run(
-                [sys.executable, "-c", "import torch; print(getattr(torch.version,'hip','') or '')"],
+                [
+                    sys.executable,
+                    "-c",
+                    "import torch; print(getattr(torch.version,'hip','') or '')",
+                ],
                 stdout = subprocess.PIPE,
                 stderr = subprocess.DEVNULL,
                 timeout = 30,
@@ -289,8 +293,13 @@ def _ensure_rocm_torch() -> None:
             print("   ROCm detected but version unreadable -- skipping torch reinstall")
             return
         entry = next(
-            ((rt, tv) for (maj, mn), (rt, tv) in sorted(_ROCM_WINDOWS_RELEASES.items(), reverse = True)
-             if ver >= (maj, mn)),
+            (
+                (rt, tv)
+                for (maj, mn), (rt, tv) in sorted(
+                    _ROCM_WINDOWS_RELEASES.items(), reverse = True
+                )
+                if ver >= (maj, mn)
+            ),
             None,
         )
         if entry is None:
@@ -303,7 +312,9 @@ def _ensure_rocm_torch() -> None:
             f"{_ROCM_WINDOWS_WHEEL_BASE}/{rel_tag}/"
             f"torch-{torch_ver}-cp312-cp312-win_amd64.whl"
         )
-        print(f"   ROCm {ver[0]}.{ver[1]} (Windows) -- installing torch from {wheel_url}")
+        print(
+            f"   ROCm {ver[0]}.{ver[1]} (Windows) -- installing torch from {wheel_url}"
+        )
         pip_install(
             f"ROCm torch (Windows, {rel_tag})",
             "--force-reinstall",