diff --git a/tests/trainer/test_fsdp_checkpoint.py b/tests/trainer/test_fsdp_checkpoint.py index 5f97c6092c..053176cf05 100644 --- a/tests/trainer/test_fsdp_checkpoint.py +++ b/tests/trainer/test_fsdp_checkpoint.py @@ -513,6 +513,7 @@ def test_fsdp_mixed_with_sync( '0.22.0', '0.23.0', '0.24.0', + '0.25.0', ], ) @pytest.mark.filterwarnings(r'ignore:.*metrics are not saved with sharded state dict.*:UserWarning')