Update tests/plugins/test_deepspeed_plugin.py

Sounds good Co-authored-by: Sean Naren <sean@grid.ai>
Lightning-AI · Dec 7, 2021 · 55b2b45 · 55b2b45
1 parent cdbebe6
commit 55b2b45
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/plugins/test_deepspeed_plugin.py b/tests/plugins/test_deepspeed_plugin.py
@@ -361,7 +361,7 @@ def test_deepspeed_custom_activation_checkpointing_params(tmpdir):
     assert checkpoint_config["synchronize_checkpoint_boundary"]
 
 
-@RunIf(min_gpus=1, deepspeed=True)
+@RunIf(min_gpus=1, deepspeed=True, standalone=True)
 def test_deepspeed_custom_activation_checkpointing_params_forwarded(tmpdir):
     """Ensure if we modify the activation checkpointing parameters, we pass these to
     deepspeed.checkpointing.configure correctly."""