Apply suggestions from code review

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
Lightning-AI · May 12, 2020 · a44f54d · a44f54d
1 parent 6578951
commit a44f54d
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/pytorch_lightning/core/saving.py b/pytorch_lightning/core/saving.py
@@ -52,11 +52,13 @@ def update_hparams(hparams, updates):
     """
     Overrides hparams with new values
 
-    >>> hparams = {'a', {'b': 2}, 'c': 1}
+    >>> hparams = {'a': {'b': 2}, 'c': 1}
     >>> updates = {'a': {'b': 4}, 'c': 7}
     >>> update_hparams(hparams, updates)
-    >>> assert hparams['a']['b'] == 4, 'update hparams failed'
-    >>> assert hparams['c'] == 7, 'update hparams failed'
+    >>> hparams['a']['b']
+    4
+    >>> hparams['c']
+    7
 
     Args:
         hparams: