Apply suggestions from code review

Lightning-AI · Mar 13, 2021 · a5cb115 · a5cb115
1 parent e29f6e3
commit a5cb115
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/docs/source/pages/overview.rst b/docs/source/pages/overview.rst
@@ -195,7 +195,7 @@ Example:
 Similarly it can also reduce the amount of code required to log multiple metrics
 inside your LightningModule
 
-.. code-block:: python
+.. testcode::
 
     from torchmetrics import Accuracy, MetricCollection, Precision, Recall
 
@@ -206,15 +206,15 @@ inside your LightningModule
 
     def training_step(self, batch, batch_idx):
         logits = self(x)
-        ...
+        # ...
         output = self.train_metrics(logits, y)
         # use log_dict instead of log
         # metrics are logged with keys: train_Accuracy, train_Precision and train_Recall
         self.log_dict(output)
 
     def validation_step(self, batch, batch_idx):
         logits = self(x)
-        ...
+        # ...
         # use log_dict instead of log
         # metrics are logged with keys: val_Accuracy, val_Precision and val_Recall
         self.log_dict(output)