Skip to content

Commit 1f98f18

Browse files
AMHermansenlantiga
authored andcommitted
Fix typo (#18986)
(cherry picked from commit ebc7cc3)
1 parent a49b1c8 commit 1f98f18

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/lightning/pytorch/callbacks/prediction_writer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ def __init__(self, output_dir, write_interval):
5858
self.output_dir = output_dir
5959
6060
def write_on_batch_end(
61-
self, trainer, pl_module', prediction, batch_indices, batch, batch_idx, dataloader_idx
61+
self, trainer, pl_module, prediction, batch_indices, batch, batch_idx, dataloader_idx
6262
):
6363
torch.save(prediction, os.path.join(self.output_dir, dataloader_idx, f"{batch_idx}.pt"))
6464

0 commit comments

Comments
 (0)