def _flatten_nested()

in src/chug/wds/filters.py [0:0]

• 10 lines of code
• 7 McCabe index (conditional complexity)

def _flatten_nested(data, *args, replace_existing=True, remove_original=True):
    """Convert dict samples to tuples."""
    for sample in data:
        for k in args:
            nested_dict = sample.pop(k, {}) if remove_original else sample.get(k, {})
            if replace_existing:
                sample.update(nested_dict)
            elif k in sample:
                for sk, sv in nested_dict.items():
                    sample.setdefault(sk, sv)
        yield sample