also update varargs passed in

Lightning-AI · Jun 2, 2020 · 3d7a30d · 3d7a30d
1 parent 9fca7a8
commit 3d7a30d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pytorch_lightning/core/lightning.py b/pytorch_lightning/core/lightning.py
@@ -1762,6 +1762,7 @@ def _collect_init_args(frame, path_args: list) -> list:
 
         # only collect variables that appear in the signature
         local_args = {k: local_vars[k] for k in init_parameters.keys()}
+        local_args.update(local_args.get(varargs_identifier, {}))
         local_args.update(local_args.get(kwargs_identifier, {}))
         local_args = {k: v for k, v in local_args.items() if k not in exclude_argnames}