[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
997155d193
commit
0c0e2cbbc7
1 changed files with 3 additions and 1 deletions
|
|
@ -701,7 +701,9 @@ class TestEnsureRocmTorch:
|
|||
with patch("os.path.isdir", return_value = True):
|
||||
with patch("subprocess.run", return_value = mock_probe):
|
||||
# Would raise if the Strix block ran (it is skipped on an explicit pin).
|
||||
with patch.object(stack_mod, "_detect_amd_gfx_codes", side_effect = AssertionError):
|
||||
with patch.object(
|
||||
stack_mod, "_detect_amd_gfx_codes", side_effect = AssertionError
|
||||
):
|
||||
_ensure_rocm_torch()
|
||||
assert mock_pip.call_count == 1
|
||||
torch_call = str(mock_pip.call_args_list[0])
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue