Merge pull request #18 from predictive-analytics-lab/flatten-dict

flatten dict

kit/__init__.py

@@ -1,3 +1,4 @@
 from .hydra import *
 from .logging import *
+from .misc import *
 from .typing import *

kit/hydra.py

@@ -15,25 +15,12 @@
 __all__ = [
     "GroupRegistration",
     "SchemaRegistration",
-    "flatten_dict",
     "as_pretty_dict",
     "reconstruct_cmd",
     "recursively_instantiate",
 ]
 
 
-def flatten_dict(d: MutableMapping, parent_key: str = "", sep: str = ".") -> dict:
-    """Flatten a nested dictionary by separating the keys with `sep`."""
-    items: list[tuple[Any, Any]] = []
-    for k, v in d.items():
-        new_key = parent_key + sep + k if parent_key else k
-        if isinstance(v, MutableMapping):
-            items.extend(flatten_dict(v, new_key, sep=sep).items())
-        else:
-            items.append((new_key, v))
-    return dict(items)
-
-
 def _clean_up_dict(obj: Any) -> Any:
     """Convert enums to strings and filter out _target_."""
     if isinstance(obj, MutableMapping):

kit/misc.py

@@ -0,0 +1,18 @@
+from __future__ import annotations
+from typing import Any, MutableMapping
+
+__all__ = ["flatten_dict"]
+
+
+def flatten_dict(
+    d: MutableMapping[str, Any], parent_key: str = "", sep: str = "."
+) -> dict[str, Any]:
+    """Flatten a nested dictionary by separating the keys with `sep`."""
+    items = []
+    for k, v in d.items():
+        new_key = parent_key + sep + k if parent_key else k
+        if isinstance(v, MutableMapping):
+            items.extend(flatten_dict(v, new_key, sep=sep).items())
+        else:
+            items.append((new_key, v))
+    return dict(items)

pyproject.toml

@@ -54,7 +54,7 @@ readme="README.md"
 
 [tool.poetry.dependencies]
 python = ">=3.7,<4.0"
-hydra-core = "1.1.0.dev4"
+hydra-core = "^1.1.0.dev4"
 numpy = "^1.20.1"
 torch = { version = "^1.8", optional = true }