concisely maintain backward compatibility for hparams type

Lightning-AI · Mar 17, 2020 · 65c9f84 · 65c9f84
1 parent 13874b3
commit 65c9f84
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/pytorch_lightning/core/lightning.py b/pytorch_lightning/core/lightning.py
@@ -1396,9 +1396,7 @@ def _load_model_state(cls, checkpoint: Dict[str, Any]) -> 'LightningModule':
 
         if cls_takes_hparams:
             if ckpt_hparams is not None:
-                is_namespace = False
-                if 'hparams_type' not in checkpoint or checkpoint.get('hparams_type') == 'namespace':
-                    is_namespace = True
+                is_namespace = checkpoint.get('hparams_type', 'namespace') == 'namespace'
                 hparams = Namespace(**ckpt_hparams) if is_namespace else ckpt_hparams
             else:
                 warnings.warn(