lint

Lightning-AI · Jan 29, 2024 · be2fcc0 · be2fcc0
1 parent 0a3606e
commit be2fcc0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/src/lightning_graphcore/strategy.py b/src/lightning_graphcore/strategy.py
@@ -228,9 +228,7 @@ def _convert_to_poptorch_loader(
             # the user is returning the `poptorch.DataLoader` directly, don't change anything.
             return dataloader
 
-        dl_args, dl_kwargs = _get_dataloader_init_args_and_kwargs(
-            dataloader, sampler, mode
-        )
+        dl_args, dl_kwargs = _get_dataloader_init_args_and_kwargs(dataloader, sampler, mode)
         opts = self.training_opts if mode == RunningStage.TRAINING else self.inference_opts
         return _reinstantiate_wrapped_cls(dataloader, opts, *dl_args, explicit_cls=poptorch.DataLoader, **dl_kwargs)
 

diff --git a/tests/test_accelerator.py b/tests/test_accelerator.py
@@ -81,7 +81,7 @@ def test_all_stages(tmpdir, devices):
     trainer.predict(model)
 
 
-@pytest.mark.parametrize("devices",[1, 4])
+@pytest.mark.parametrize("devices", [1, 4])
 def test_inference_only(tmpdir, devices):
     model = IPUModel()