From fbb9b8156fc9770b89236bdc7e21d63d02aeb763 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Mon, 13 Jul 2026 11:39:16 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 studio/backend/core/export/export.py        |  3 ++-
 studio/backend/tests/test_torchao_select.py | 12 +++++-------
 2 files changed, 7 insertions(+), 8 deletions(-)

diff --git a/studio/backend/core/export/export.py b/studio/backend/core/export/export.py
index 0e25bb343a..3ae214520a 100644
--- a/studio/backend/core/export/export.py
+++ b/studio/backend/core/export/export.py
@@ -109,9 +109,11 @@ def _torchao_export_supported():
     Linux, and macOS are unaffected (torchao is real)."""
     try:
         from core._torchao_stub import is_win32_rocm
+
         if is_win32_rocm():
             return False
         import unsloth.save as _us
+
         return hasattr(_us, "_normalize_torchao_method")
     except Exception:
         return False
@@ -121,7 +123,6 @@ def _torchao_runtime_unavailable():
     """True where portable torchao export cannot run (Windows ROCm): torchao is import-stubbed
     (its config classes return None) or torch.distributed is absent. False everywhere else."""
     import sys
-
     try:
         from core._torchao_stub import is_win32_rocm, _STUB_SENTINEL
         if is_win32_rocm():
diff --git a/studio/backend/tests/test_torchao_select.py b/studio/backend/tests/test_torchao_select.py
index fe5100c7d4..3b8b0e6ffd 100644
--- a/studio/backend/tests/test_torchao_select.py
+++ b/studio/backend/tests/test_torchao_select.py
@@ -173,17 +173,15 @@ def _exec_func(rel, name):
 @pytest.mark.parametrize(
     ("platform", "hip", "version", "expected"),
     [
-        ("win32", "6.4.0", "2.10.0+rocm6.4", True),   # ROCm via torch.version.hip
-        ("win32", None, "2.10.0+rocm6.4", True),      # ROCm via __version__ tag only
-        ("win32", None, "2.10.0+cu128", False),       # Windows CUDA -> real torchao
+        ("win32", "6.4.0", "2.10.0+rocm6.4", True),  # ROCm via torch.version.hip
+        ("win32", None, "2.10.0+rocm6.4", True),  # ROCm via __version__ tag only
+        ("win32", None, "2.10.0+cu128", False),  # Windows CUDA -> real torchao
         ("linux", "6.4.0", "2.10.0+rocm6.4", False),  # Linux ROCm -> real torchao
-        ("darwin", None, "2.10.0", False),            # macOS
+        ("darwin", None, "2.10.0", False),  # macOS
     ],
 )
 def test_is_win32_rocm(monkeypatch, platform, hip, version, expected):
-    fake_torch = types.SimpleNamespace(
-        version = types.SimpleNamespace(hip = hip), __version__ = version
-    )
+    fake_torch = types.SimpleNamespace(version = types.SimpleNamespace(hip = hip), __version__ = version)
     monkeypatch.setattr(sys, "platform", platform)
     monkeypatch.setitem(sys.modules, "torch", fake_torch)
     assert _stub.is_win32_rocm() is expected