Fix typo in BasePredictionWriter documentation (#18381)

Lightning-AI · Aug 24, 2023 · 9d7a284 · 9d7a284
1 parent 9496d9a
commit 9d7a284
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docs/source-pytorch/deploy/production_basic.rst b/docs/source-pytorch/deploy/production_basic.rst
@@ -94,7 +94,7 @@ By using the predict step in Lightning you get free distributed inference using
             torch.save(batch_indices, os.path.join(self.output_dir, f"batch_indices_{trainer.global_rank}.pt"))
 
 
-    # or you can set `writer_interval="batch"` and override `write_on_batch_end` to save
+    # or you can set `write_interval="batch"` and override `write_on_batch_end` to save
     # predictions at batch level
     pred_writer = CustomWriter(output_dir="pred_path", write_interval="epoch")
     trainer = Trainer(accelerator="gpu", strategy="ddp", devices=8, callbacks=[pred_writer])

diff --git a/src/lightning/pytorch/callbacks/prediction_writer.py b/src/lightning/pytorch/callbacks/prediction_writer.py
@@ -93,7 +93,7 @@ def write_on_epoch_end(self, trainer, pl_module, predictions, batch_indices):
                 torch.save(batch_indices, os.path.join(self.output_dir, f"batch_indices_{trainer.global_rank}.pt"))
 
 
-        # or you can set `writer_interval="batch"` and override `write_on_batch_end` to save
+        # or you can set `write_interval="batch"` and override `write_on_batch_end` to save
         # predictions at batch level
         pred_writer = CustomWriter(output_dir="pred_path", write_interval="epoch")
         trainer = Trainer(accelerator="gpu", strategy="ddp", devices=8, callbacks=[pred_writer])