pep8

Lightning-AI · Aug 4, 2020 · 74815d9 · 74815d9
1 parent 4716885
commit 74815d9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/trainer/test_dataloaders.py b/tests/trainer/test_dataloaders.py
@@ -263,7 +263,8 @@ def test_multiple_dataloaders_passed_to_fit(tmpdir, ckpt_path):
         pytest.param(1.0, 1.0, 1.0),
     ]
 )
-def test_inf_dataloaders_with_limit_percent_batches(tmpdir, limit_train_batches, limit_val_batches, limit_test_batches):
+def test_inf_dataloaders_with_limit_percent_batches(tmpdir, limit_train_batches,
+                                                    limit_val_batches, limit_test_batches):
     """Verify inf train, val & test dataloaders (e.g. IterableDataset) passed with batch limit in percent"""
     model = EvalModelTemplate()
     model.train_dataloader = model.train_dataloader__infinite