diff --git a/docs/source-pytorch/extensions/logging.rst b/docs/source-pytorch/extensions/logging.rst
index 6a2422f295..f0c12464e6 100644
--- a/docs/source-pytorch/extensions/logging.rst
+++ b/docs/source-pytorch/extensions/logging.rst
@@ -326,7 +326,7 @@ if you are using a logger. These defaults can be customized by overriding the
     class CustomProgressBar(TQDMProgressBar):
         def get_metrics(self, *args, **kwargs):
             # don't show the version number
-            items = super().get_metrics()
+            items = super().get_metrics(*args, **kwargs)
             items.pop("v_num", None)
             return items