Fix l2l.utils.flatten_config.

learnables · Jun 20, 2023 · e56657c · e56657c
1 parent 2cdfeed
commit e56657c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/learn2learn/utils/__init__.py b/learn2learn/utils/__init__.py
@@ -337,10 +337,10 @@ def flatten_config(args, prefix=None):
         return flat_args
     elif dataclasses.is_dataclass(args):
         keys = dataclasses.fields(args)
-        def getvalue(x): getattr(args, x.name)
+        def getvalue(x): return getattr(args, x.name)
     elif isinstance(args, dict):
         keys = args.keys()
-        def getvalue(x): args[x]
+        def getvalue(x): return args[x]
     else:
         raise 'Unknown args'
     for key in keys: