[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

tests/tests_pytorch/helpers/advanced_models.py

@@ -218,4 +218,9 @@ def configure_optimizers(self):
         return torch.optim.Adam(self.parameters(), lr=0.02)
 
     def train_dataloader(self):
-        return DataLoader(MNIST(root=_PATH_DATASETS, train=True, download=True), batch_size=128, num_workers=1, persistent_workers=True)
+        return DataLoader(
+            MNIST(root=_PATH_DATASETS, train=True, download=True),
+            batch_size=128,
+            num_workers=1,
+            persistent_workers=True,
+        )

tests/tests_pytorch/trainer/connectors/test_data_connector.py

@@ -252,7 +252,9 @@ def test_update_dataloader_with_multiprocessing_context():
     """This test verifies that `use_distributed_sampler` conserves multiprocessing context."""
     train = RandomDataset(32, 64)
     context = "spawn"
-    train = DataLoader(train, batch_size=32, num_workers=2, multiprocessing_context=context, shuffle=True, persistent_workers=True)
+    train = DataLoader(
+        train, batch_size=32, num_workers=2, multiprocessing_context=context, shuffle=True, persistent_workers=True
+    )
     new_data_loader = _update_dataloader(train, SequentialSampler(train.dataset))
     assert new_data_loader.multiprocessing_context == train.multiprocessing_context