From 110e5afb4b83db183e2ba9f7d452ac9fb8fb3703 Mon Sep 17 00:00:00 2001 From: Remi Date: Mon, 22 Apr 2024 14:01:57 +0200 Subject: [PATCH] Update lerobot/common/datasets/utils.py Co-authored-by: Alexander Soare --- lerobot/common/datasets/utils.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/lerobot/common/datasets/utils.py b/lerobot/common/datasets/utils.py index 741082ef..6c99ffa5 100644 --- a/lerobot/common/datasets/utils.py +++ b/lerobot/common/datasets/utils.py @@ -16,6 +16,14 @@ from lerobot.common.utils.utils import set_global_seed def flatten_dict(d, parent_key="", sep="/"): + """Flatten a nested dictionary structure by collapsing nested keys into one key with a separator. + + For example: + ``` + >>> dct = {"a": {"b": 1, "c": {"d": 2}}, "e": 3}` + >>> print(flatten_dict(dct)) + {"a/b": 1, "a/c/d": 2, "e": 3} + """ items = [] for k, v in d.items(): new_key = f"{parent_key}{sep}{k}" if parent_key else k