add 0.17 to bcompat tests (#2723)

mosaicml · Nov 17, 2023 · 1a4e67f · 1a4e67f
1 parent ea2ceff
commit 1a4e67f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/trainer/test_fsdp_checkpoint.py b/tests/trainer/test_fsdp_checkpoint.py
@@ -393,6 +393,11 @@ def test_fsdp_mixed_with_sync(
         '0.16.0',
         marks=pytest.mark.filterwarnings((r'ignore:MosaicMLLogger is not in the state_dict. Its '
                                           r'state will not be restored.:UserWarning')),
+    ),
+    pytest.param(
+        '0.17.0',
+        marks=pytest.mark.filterwarnings((r'ignore:MosaicMLLogger is not in the state_dict. Its '
+                                          r'state will not be restored.:UserWarning')),
     )
 ])
 @pytest.mark.filterwarnings(r'ignore:.*metrics are not saved with sharded state dict.*:UserWarning')