Skip to content

Commit a905095

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent abac98f commit a905095

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/lightning/pytorch/loops/prediction_loop.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -233,7 +233,9 @@ def _predict_step(
233233

234234
self.batch_progress.increment_ready()
235235

236-
any_on_epoch = self._store_data_for_prediction_writer(batch_idx, dataloader_idx) if not using_dataloader_iter else False
236+
any_on_epoch = (
237+
self._store_data_for_prediction_writer(batch_idx, dataloader_idx) if not using_dataloader_iter else False
238+
)
237239

238240
# the `_step` methods don't take a batch_idx when `dataloader_iter` is used, but all other hooks still do,
239241
# so we need different kwargs

0 commit comments

Comments
 (0)