diff --git a/tests/plugins/test_ddp_plugin.py b/tests/plugins/test_ddp_plugin.py index 22f7da667686bf..ca5f76df14356a 100644 --- 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)