Update src/lightning/pytorch/cli.py

Co-authored-by: Nicki Skafte Detlefsen <[email protected]>

Author: mauvilsa

src/lightning/pytorch/cli.py

@@ -566,8 +566,8 @@ def _parse_ckpt_path(self) -> None:
                 return
             if "_class_path" in hparams:
                 hparams = {
-                    "class_path": hparams["_class_path"],
-                    "dict_kwargs": {k: v for k, v in hparams.items() if k != "_class_path"},
+                    "class_path": hparams.pop("_class_path"),
+                    "dict_kwargs": hparams,
                 }
             hparams = {self.config.subcommand: {"model": hparams}}
             try: