diff --git a/tests/accelerators/test_dp.py b/tests/accelerators/test_dp.py index a094cfd380cc6..1d79ff8c59eba 100644 --- 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)