def _disable_decode()

in src/chug/hfds/loader.py [0:0]

• 15 lines of code
• 5 McCabe index (conditional complexity)

def _disable_decode(ds):
    import datasets

    to_set = []
    for k, v in ds.features.items():
        if isinstance(v, datasets.Image):
            d = deepcopy(v)
            d.decode = False
            to_set.append((k, d))
        elif isinstance(v, datasets.Audio):
            d = deepcopy(v)
            d.decode = False
            to_set.append((k, d))
    for k, d in to_set:
        ds = ds.cast_column(k, d)
    return ds