fix name in test

Lightning-AI · Dec 20, 2021 · 6063f70 · 6063f70
1 parent 3d11bea
commit 6063f70
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/plugins/test_ddp_plugin.py b/tests/plugins/test_ddp_plugin.py
@@ -113,9 +113,9 @@ def test_ddp_configure_ddp():
     trainer.strategy.setup_environment()
     trainer.strategy.setup(trainer)
     trainer.lightning_module.trainer = trainer
-    trainer.training_type_plugin.setup_environment()
+    trainer.strategy.setup_environment()
     assert isinstance(trainer.model, LightningModule)
-    trainer.training_type_plugin.setup(trainer)
+    trainer.strategy.setup(trainer)
     # in DDPPlugin configure_ddp(), model wrapped by DistributedDataParallel
     assert isinstance(trainer.model, DistributedDataParallel)
 
@@ -125,9 +125,9 @@ def test_ddp_configure_ddp():
     )
     # test do not wrap the model if trainerFN is not fitting
     trainer.state.fn = TrainerFn.VALIDATING
-    trainer.training_type_plugin.connect(model)
+    trainer.strategy.connect(model)
     trainer.lightning_module.trainer = trainer
-    trainer.training_type_plugin.setup_environment()
-    trainer.training_type_plugin.setup(trainer)
+    trainer.strategy.setup_environment()
+    trainer.strategy.setup(trainer)
     # in DDPPlugin configure_ddp(), model are still LightningModule
     assert isinstance(trainer.model, LightningModule)