Update tests/trainer/test_fsdp_checkpoint.py

fix formatting Co-authored-by: Mihir Patel <mihir.v.patel7@gmail.com>
mosaicml · Jun 24, 2024 · dd64f47 · dd64f47
1 parent 95be0e3
commit dd64f47
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/trainer/test_fsdp_checkpoint.py b/tests/trainer/test_fsdp_checkpoint.py
@@ -806,7 +806,7 @@ def mock_get_checkpoint_validation_function():
             False,
             False,
             False,
-            marks=pytest.mark.world_size(2),
+            marks=pytest.mark.world_size(2)
         ),
         pytest.param(False, 'adamw', 'amp_bf16', False, None, True, False, False, marks=pytest.mark.world_size(2)),
         pytest.param(False, 'adamw', 'amp_bf16', False, None, False, True, False, marks=pytest.mark.world_size(4)),