Apply suggestions from code review

Lightning-AI · Jan 29, 2021 · 238e0f2 · 238e0f2
1 parent 8655c8c
commit 238e0f2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)