Update tests/accelerators/test_dp.py

Co-authored-by: Carlos Mocholí <[email protected]>

Author: tchaton

tests/accelerators/test_dp.py

@@ -130,13 +130,7 @@ def test_dp_training_step_dict(tmpdir):
     """
     This test verify dp properly reduce dictionaries
     """
-
-    class TestModel(BoringModel):
-
-        def training_step(self, batch, batch_idx):
-            return super().training_step(batch, batch_idx)
-
-    model = TestModel()
+    model = BoringModel()
     model.training_step_end = None
     trainer = pl.Trainer(
         default_root_dir=tmpdir,