ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix IS_ROCM test assertion for multi-line formatting

Browse source
This commit is contained in:
Daniel Han 2026-03-31 11:58:49 +00:00
commit 478bc7f625
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/studio/install/test_rocm_support.py
View file

@ -741,7 +741,7 @@ class TestHardwareRocmFlag:
PACKAGE_ROOT / "studio" / "backend" / "utils" / "hardware" / "hardware.py"
)
source = hw_path.read_text()
assert "IS_ROCM: bool = False" in source
assert "IS_ROCM: bool" in source and "False" in source
def test_hardware_py_sets_is_rocm_on_hip(self):
"""detect_hardware() should set IS_ROCM when torch.version.hip is set."""