From 238e0f2457707755fa1c8432ec961b9308afa376 Mon Sep 17 00:00:00 2001 From: Jirka Borovec Date: Fri, 29 Jan 2021 10:14:48 +0100 Subject: [PATCH] Apply suggestions from code review --- tests/deprecated_api/test_remove_1-4.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/deprecated_api/test_remove_1-4.py b/tests/deprecated_api/test_remove_1-4.py index 38e02c9187159..27af0003beb43 100644 --- a/tests/deprecated_api/test_remove_1-4.py +++ b/tests/deprecated_api/test_remove_1-4.py @@ -196,5 +196,5 @@ def test_v1_4_0_deprecated_lightning_data_parallel(): match="`LightningDataParallel` is deprecated since v1.2 and will be removed in v1.4." ): dp_model = LightningDataParallel(model, device_ids=[0]) - assert isinstance(dp_model, torch.nn.DataParallel) - assert isinstance(dp_model.module, LightningParallelModule) + assert isinstance(dp_model, torch.nn.DataParallel) + assert isinstance(dp_model.module, LightningParallelModule)