From 9b1e45f58bc0510b845fe3f98f67e8d9ef087f91 Mon Sep 17 00:00:00 2001
From: thomas chaton <thomas@grid.ai>
Date: Thu, 4 Mar 2021 12:57:44 +0000
Subject: [PATCH] Update tests/accelerators/test_dp.py

Co-authored-by: Kaushik B <45285388+kaushikb11@users.noreply.github.com>
---
 tests/accelerators/test_dp.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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)