Update tests/plugins/test_ddp_module_wrapper.py

replace "ddp" with "ddp_cpu" Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
Lightning-AI · Aug 12, 2021 · 3869780 · 3869780
1 parent 60f0ee9
commit 3869780
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/plugins/test_ddp_module_wrapper.py b/tests/plugins/test_ddp_module_wrapper.py
@@ -55,7 +55,7 @@ def on_predict_start(self) -> None:
 @pytest.mark.parametrize(
     ["accelerator"],
     [
-        ("ddp",),
+        ("ddp_cpu",),
         ("ddp_spawn",),
     ],
 )