Skip to content

Commit 83436ee

Browse files
Fixing small typo in documentation. (Lightning-AI#13215)
Fixing typo in documentation.
1 parent 7938293 commit 83436ee

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytorch_lightning/callbacks/prediction_writer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ class CustomWriter(BasePredictionWriter):
5454
5555
def __init__(self, output_dir: str, write_interval: str):
5656
super().__init__(write_interval)
57-
self.output_dir
57+
self.output_dir = output_dir
5858
5959
def write_on_batch_end(
6060
self, trainer, pl_module: 'LightningModule', prediction: Any, batch_indices: List[int], batch: Any,

0 commit comments

Comments
 (0)