From 3067860f755466442d15fe4521c9531a0318788e Mon Sep 17 00:00:00 2001 From: Carlos Mocholi Date: Wed, 30 Jun 2021 00:28:43 +0200 Subject: [PATCH] Use DDPPlugin --- tests/deprecated_api/test_remove_1-6.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tests/deprecated_api/test_remove_1-6.py b/tests/deprecated_api/test_remove_1-6.py index f41c2e9218693..5fe6bb210dab7 100644 --- a/tests/deprecated_api/test_remove_1-6.py +++ b/tests/deprecated_api/test_remove_1-6.py @@ -16,7 +16,6 @@ from pytorch_lightning import Trainer from pytorch_lightning.callbacks.early_stopping import EarlyStopping -from pytorch_lightning.plugins import DeepSpeedPlugin from pytorch_lightning.plugins.training_type import DDPPlugin, DDPSpawnPlugin from pytorch_lightning.utilities.distributed import rank_zero_deprecation, rank_zero_warn from pytorch_lightning.utilities.model_helpers import is_overridden @@ -247,6 +246,6 @@ def test_v1_6_0_rank_zero_warnings_moved(): def test_v1_6_0_ddp_plugin_task_idx(): - plugin = DeepSpeedPlugin() - with pytest.deprecated_call(match='Use `DeepSpeedPlugin.local_rank` instead'): + plugin = DDPPlugin() + with pytest.deprecated_call(match='Use `DDPPlugin.local_rank` instead'): _ = plugin.task_idx