We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent abac98f commit a905095Copy full SHA for a905095
src/lightning/pytorch/loops/prediction_loop.py
@@ -233,7 +233,9 @@ def _predict_step(
233
234
self.batch_progress.increment_ready()
235
236
- any_on_epoch = self._store_data_for_prediction_writer(batch_idx, dataloader_idx) if not using_dataloader_iter else False
+ any_on_epoch = (
237
+ self._store_data_for_prediction_writer(batch_idx, dataloader_idx) if not using_dataloader_iter else False
238
+ )
239
240
# the `_step` methods don't take a batch_idx when `dataloader_iter` is used, but all other hooks still do,
241
# so we need different kwargs
0 commit comments