Update tests/accelerators/test_dp.py

Co-authored-by: Kaushik B <45285388+kaushikb11@users.noreply.github.com>
Lightning-AI · Mar 4, 2021 · 9b1e45f · 9b1e45f
1 parent 6b3d47e
commit 9b1e45f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/accelerators/test_dp.py b/tests/accelerators/test_dp.py
@@ -138,6 +138,6 @@ def test_dp_training_step_dict(tmpdir):
         limit_train_batches=2,
         limit_val_batches=0,
         gpus=2,
-        plugins='dp',
+        accelerator='dp',
     )
     trainer.fit(model)