fix test

Lightning-AI · Jan 27, 2021 · 2b0b465 · 2b0b465
1 parent ee16408
commit 2b0b465
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/overrides/test_data_parallel.py b/tests/overrides/test_data_parallel.py
@@ -30,7 +30,7 @@ def test_lightning_distributed_module_methods():
     pl_module.validation_step.assert_called_with(batch, batch_idx)
 
 
-def test_lightning_distributed_module_warn_none_output(wrapper_class):
+def test_lightning_distributed_module_warn_none_output():
     """ Test that the LightningModuleWrapper warns about forgotten return statement. """
     pl_module = MagicMock()
     wrapped_module = LightningModuleWrapperBase(pl_module)