yo

mosaicml · Dec 3, 2024 · 790e25e · 790e25e
1 parent b29d4ed
commit 790e25e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_events.py b/tests/test_events.py
@@ -80,7 +80,7 @@ def get_trainer(self, precision='fp32', max_duration='1ep', save_interval='1ep',
     @pytest.mark.parametrize(
         'device,use_fsdp,precision',
         [
-            pytest.param('cpu', None, False, 'fp32', id='cpu-ddp'),
+            pytest.param('cpu', False, 'fp32', id='cpu-ddp'),
             # TODO: Remove filterwarnings after FSDP remove deprecated code
             pytest.param(
                 'gpu',