Skip to content

Commit 7fa6973

Browse files
Another test gaurd for torch >= 2.5
1 parent 62267d7 commit 7fa6973

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/test_parametrize.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -293,6 +293,7 @@ def test_quant_state_preservation(device, dtype):
293293

294294
@pytest.mark.parametrize("device", get_available_devices())
295295
@pytest.mark.parametrize("dtype", [torch.float32, torch.float16, torch.bfloat16], ids=describe_dtype)
296+
@pytest.mark.skipif(torch.__version__ < (2, 5), reason="state dict hook requires torch >= 2.5.0")
296297
def test_multiple_parameters(device, dtype):
297298
"""Test applying parametrization to multiple parameters in the same module."""
298299
if device == "hpu" and not is_supported_on_hpu("nf4", dtype):

0 commit comments

Comments
 (0)