[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

studio/install_python_stack.py

@@ -1926,9 +1926,9 @@ def _ensure_rocm_torch() -> None:
     # MI50 / Radeon VII path; it must win over the Strix probe-order detection
     # below (a mixed Strix + MI50 host could otherwise route to gfx1151), so the
     # Strix override is skipped when it is set.
-    _gfx906_arch_override = (
-        os.environ.get("UNSLOTH_ROCM_GFX_ARCH") or ""
-    ).strip().lower().split(":")[0] == "gfx906"
+    _gfx906_arch_override = (os.environ.get("UNSLOTH_ROCM_GFX_ARCH") or "").strip().lower().split(
+        ":"
+    )[0] == "gfx906"
 
     # Strix Halo / Point (gfx1151 / gfx1150) need torch from AMD's per-gfx index
     # (2.11+rocm7.13); any generic pytorch.org rocm index lacks the fixes (ROCm 7.1

tests/studio/install/test_rocm_support.py

@@ -838,9 +838,7 @@ def test_explicit_gfx_index_honored_and_skips_strix_reroute(
             stack_mod.os.environ.pop("UNSLOTH_TORCH_INDEX_FAMILY", None)
             with patch("os.path.isdir", return_value = True):
                 with patch("subprocess.run", return_value = mock_probe):
-                    with patch.object(
-                        stack_mod, "_detect_amd_gfx_codes", return_value = ["gfx1151"]
-                    ):
+                    with patch.object(stack_mod, "_detect_amd_gfx_codes", return_value = ["gfx1151"]):
                         _ensure_rocm_torch()
         assert mock_pip.call_count == 1
         torch_call = str(mock_pip.call_args_list[0])
@@ -942,9 +940,7 @@ def test_gfx_pin_over_installed_pre211_rocm_reinstalls(
             stack_mod.os.environ.pop("UNSLOTH_TORCH_INDEX_FAMILY", None)
             with patch("os.path.isdir", return_value = True):
                 with patch("subprocess.run", return_value = mock_probe):
-                    with patch.object(
-                        stack_mod, "_detect_amd_gfx_codes", return_value = ["gfx1151"]
-                    ):
+                    with patch.object(stack_mod, "_detect_amd_gfx_codes", return_value = ["gfx1151"]):
                         _ensure_rocm_torch()
         torch_call = str(mock_pip.call_args_list[0])
         assert "gfx1151" in torch_call
@@ -1031,9 +1027,7 @@ def test_gfx_pin_over_generic_rocm211_reinstalls(
                 with patch("subprocess.run", return_value = mock_probe):
                     # The gfx probe may run for the bnb-skip flag; returning a Strix
                     # arch must not reroute the pinned torch index.
-                    with patch.object(
-                        stack_mod, "_detect_amd_gfx_codes", return_value = ["gfx1151"]
-                    ):
+                    with patch.object(stack_mod, "_detect_amd_gfx_codes", return_value = ["gfx1151"]):
                         _ensure_rocm_torch()
         torch_call = str(mock_pip.call_args_list[0])
         assert "gfx1151" in torch_call