fix final expansion of retrieve

pesser · Aug 7, 2019 · 15a1ed4 · 15a1ed4
1 parent 1eac53a
commit 15a1ed4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/edflow/util.py b/edflow/util.py
@@ -202,6 +202,10 @@ def retrieve(
                 raise KeyNotFoundError(e)
 
             visited += [key]
+        # final expansion of retrieved value
+        if expand and callable(list_or_dict):
+            list_or_dict = list_or_dict()
+            parent[last_key] = list_or_dict
     except KeyNotFoundError as e:
         if default is None:
             print("Key not found: {}, seen: {}".format(keys, visited))