Function input_param_data

dali/test/python/sequences_test_utils.py:349–355

(arg_cb)

def compute_input_params_data(input_data: ArgData, rng, input_params: List[ArgCb]):
    def input_param_data(arg_cb):
        assert arg_cb.desc.expandable_prefix in ["", "F"]
        if arg_cb.desc.expandable_prefix == "F":
            return get_input_arg_per_frame(
                input_data, arg_cb.cb, rng, not arg_cb.desc.is_positional_arg
            )
        return get_input_arg_per_sample(input_data, arg_cb.cb, rng)

    return [ArgData(desc=arg_cb.desc, data=input_param_data(arg_cb)) for arg_cb in input_params]