Update tests/accelerators/test_dp.py

tchaton · kaushikb11 · web-flow · commit ab8a94249c3f · 2021-03-04T12:57:44.000Z
Co-authored-by: Kaushik B &lt;45285388+kaushikb11@users.noreply.github.com&gt;
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)

Original file line number	Diff line number	Diff line change
`@@ -138,6 +138,6 @@ def test_dp_training_step_dict(tmpdir):`
`138`	`138`	`limit_train_batches=2,`
`139`	`139`	`limit_val_batches=0,`
`140`	`140`	`gpus=2,`
`141`		`- plugins='dp',`
	`141`	`+ accelerator='dp',`
`142`	`142`	`)`
`143`	`143`	`trainer.fit(model)`