resolve tests

Lightning-AI · Apr 20, 2021 · 04e0cb2 · 04e0cb2
1 parent 57e006c
commit 04e0cb2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pytorch_lightning/core/lightning.py b/pytorch_lightning/core/lightning.py
@@ -1645,6 +1645,9 @@ class ``__init__`` to be ignored
  "arg1": 1
  "arg3": 3.14
  """
+ # the frame needs to be create in this file.
+ if not frame:
+ frame = inspect.currentframe().f_back
  save_hyperparameters(self, *args, ignore=ignore, frame=frame)
 
  def _set_hparams(self, hp: Union[dict, Namespace, str]) -> None: