From f44dfb3e7a88bdc55db7ebcd8c271936afcb9753 Mon Sep 17 00:00:00 2001
From: Shikhar Chauhan <xssChauhan@users.noreply.github.com>
Date: Mon, 17 Feb 2020 14:10:10 +0100
Subject: [PATCH] Fixing Function Signatures (#871)

---
 docs/source/experiment_reporting.rst | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/docs/source/experiment_reporting.rst b/docs/source/experiment_reporting.rst
index 5f61c6b93c..58a865d4e3 100644
--- a/docs/source/experiment_reporting.rst
+++ b/docs/source/experiment_reporting.rst
@@ -38,7 +38,7 @@ To plot metrics into whatever logger you passed in (tensorboard, comet, neptune,
 
 .. code-block:: python
 
-   def training_end(self, batch, batch_idx):
+   def training_end(self, outputs):
       loss = some_loss()
       ...
 
@@ -46,7 +46,7 @@ To plot metrics into whatever logger you passed in (tensorboard, comet, neptune,
       results = {'log': logs}
       return results
 
-   def validation_end(self, batch, batch_idx):
+   def validation_end(self, outputs):
       loss = some_loss()
       ...
 
@@ -54,7 +54,7 @@ To plot metrics into whatever logger you passed in (tensorboard, comet, neptune,
       results = {'log': logs}
       return results
 
-   def test_end(self, batch, batch_idx):
+   def test_end(self, outputs):
       loss = some_loss()
       ...
 
@@ -99,7 +99,7 @@ Here we show the validation loss in the progress bar
 
 .. code-block:: python
 
-   def validation_end(self, batch, batch_idx):
+   def validation_end(self, outputs):
       loss = some_loss()
       ...