o ݱi@sFddlmZddlZdeeejfdeeeeejfffddZdS))UnionNxreturncCsti}g}|D]#\}}|dr+|dd}||vs"Jd||||<||q|D]}||=q.||d<|S)a Unflattens entries with keys starting with "obs" as follows: dict_data["obs.some_name"] -> dict_data["obs"]["some_name"] Args: x: A dictionary of tensors. Returns: The same dictionary, but keys starting with "obs" are unflattened. zobs.NzDuplicate key obs)items startswithappend)rr to_deletekeyvaluenew_keyrD/data/cameron/vidgen/unified-world-model/datasets/utils/obs_utils.py unflatten_obss    r)typingrtorchdictstrTensorrrrrrs 2