From ec40e9f207a68aab42c1e2e919126204797cd1a3 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:34:34 +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 | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/studio/install_python_stack.py b/studio/install_python_stack.py
index 4b9159529d..78885bfeef 100644
--- a/studio/install_python_stack.py
+++ b/studio/install_python_stack.py
@@ -65,7 +65,12 @@ _ROCM_WINDOWS_WHEEL_BASE = (
 ).rstrip("/")
 # Maps (major, minor) → (release_folder, torch_ver, torchvision_ver, torchaudio_ver)
 _ROCM_WINDOWS_RELEASES: dict[tuple[int, int], tuple[str, str, str, str]] = {
-    (7, 2): ("rocm-rel-7.2.1", "2.9.1+rocm7.2.1", "0.24.1+rocm7.2.1", "2.9.1+rocm7.2.1"),
+    (7, 2): (
+        "rocm-rel-7.2.1",
+        "2.9.1+rocm7.2.1",
+        "0.24.1+rocm7.2.1",
+        "2.9.1+rocm7.2.1",
+    ),
 }
 
 # bitsandbytes continuous-release_main wheels with the ROCm 4-bit GEMV fix
@@ -295,9 +300,7 @@ def _ensure_rocm_torch() -> None:
         entry = next(
             (
                 v
-                for (maj, mn), v in sorted(
-                    _ROCM_WINDOWS_RELEASES.items(), reverse = True
-                )
+                for (maj, mn), v in sorted(_ROCM_WINDOWS_RELEASES.items(), reverse = True)
                 if ver >= (maj, mn)
             ),
             None,