Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2024-11-04 17:40:04 +00:00
parent 8bbe907099
commit 61cfca763c
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/tests_pytorch/callbacks/test_prediction_writer.py
View File

@ -83,7 +83,9 @@ def test_prediction_writer_batch_indices(num_workers, tmp_path):
DummyPredictionWriter.write_on_batch_end = Mock()
DummyPredictionWriter.write_on_epoch_end = Mock()
dataloader = DataLoader(RandomDataset(32, 64), batch_size=4, num_workers=num_workers, persistent_workers=num_workers > 0)
dataloader = DataLoader(
RandomDataset(32, 64), batch_size=4, num_workers=num_workers, persistent_workers=num_workers > 0
)
model = BoringModel()
writer = DummyPredictionWriter("batch_and_epoch")
trainer = Trainer(default_root_dir=tmp_path, logger=False, limit_predict_batches=4, callbacks=writer)