Function compute_input_params_data

dali/test/python/sequences_test_utils.py:348–357

(input_data: ArgData, rng, input_params: List[ArgCb])

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]


def sequence_suite_helper(rng, input_cases: List[ArgData], ops_test_cases, num_iters=4):