Function _parse_every_object

python/paddle/framework/io.py:619–652

(obj, condition_func, convert_func)

def _parse_every_object(obj, condition_func, convert_func):
    if condition_func(obj):
        return convert_func(obj)
    elif type(obj) in (dict, collections.OrderedDict, list):
        if type(obj) == list:
            keys = range(len(obj))
        else:
            keys = list(obj.keys())
        for key in keys:
            if condition_func(obj[key]):
                obj[key] = convert_func(obj[key])
            else:
                obj[key] = _parse_every_object(
                    obj[key], condition_func, convert_func
                )
        return obj
    elif type(obj) == tuple:
        return tuple(
            _parse_every_object(list(obj), condition_func, convert_func)
        )
    elif type(obj) == set:
        return set(_parse_every_object(list(obj), condition_func, convert_func))
    else:
        # Support dataclass objects - return as-is without further parsing
        if dataclasses.is_dataclass(obj):
            return obj
        if isinstance(obj, Iterable) and not isinstance(
            obj,
            (str, np.ndarray, core.eager.Tensor, core.DenseTensor),
        ):
            raise NotImplementedError(
                f"The iterable objects supported are tuple, list, dict, OrderedDict, string. But received {type(obj)}."
            )
        return obj


def _parse_load_result(obj, return_numpy):