add flatten function

mlconfig/__init__.py

@@ -1,3 +1,4 @@
+from .conf import flatten
 from .conf import getcls
 from .conf import instantiate
 from .conf import load

mlconfig/conf.py

@@ -77,3 +77,19 @@ def instantiate(conf, *args, **kwargs):
     func_or_cls = getcls(conf)
 
     return func_or_cls(*args, **kwargs)
+
+
+def flatten(data: dict, prefix: str | None = None, sep: str = ".") -> dict:
+    d = {}
+
+    for key, value in data.items():
+        if prefix is not None:
+            key = prefix + sep + key
+
+        if isinstance(value, dict):
+            d.update(flatten(value, prefix=key))
+            continue
+
+        d[key] = value
+
+    return d

tests/test_conf.py

@@ -1,5 +1,6 @@
 import pytest
 
+from mlconfig import flatten
 from mlconfig import getcls
 from mlconfig import instantiate
 from mlconfig import load
@@ -55,3 +56,8 @@ def test_instantiate(conf, obj):
 
 def test_getcls(conf):
     assert getcls(conf["a"]) == Point
+
+
+def test_flatten() -> None:
+    d = {"a": {"b": "c"}, "d": {"e": "f"}}
+    assert flatten(d) == {"a.b": "c", "d.e": "f"}