From 478bc7f625bb1b88e0fc90ae866bf4efe04a0f08 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 31 Mar 2026 11:58:49 +0000 Subject: [PATCH] Fix IS_ROCM test assertion for multi-line formatting --- tests/studio/install/test_rocm_support.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/studio/install/test_rocm_support.py b/tests/studio/install/test_rocm_support.py index 597dd8a0d9..48831fd57b 100644 --- a/tests/studio/install/test_rocm_support.py +++ b/tests/studio/install/test_rocm_support.py @@ -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."""