diff --git a/tests/studio/install/test_rocm_support.py b/tests/studio/install/test_rocm_support.py index d41adcf379..26ed005291 100644 --- a/tests/studio/install/test_rocm_support.py +++ b/tests/studio/install/test_rocm_support.py @@ -1536,7 +1536,9 @@ class TestWindowsRocmIndexUrl: assert "repo.amd.com" in url def test_mirror_env_var_overrides_base(self, monkeypatch): - monkeypatch.setenv("UNSLOTH_ROCM_WINDOWS_MIRROR", "https://my-mirror.example.com/rocm/whl") + monkeypatch.setenv( + "UNSLOTH_ROCM_WINDOWS_MIRROR", "https://my-mirror.example.com/rocm/whl" + ) # Reload module-level constant by calling helper directly url = stack_mod._windows_rocm_index_url("gfx1200") # The env var is read at module load time for _ROCM_WINDOWS_INDEX_BASE,