Function pipeline

dali/test/python/sequences_test_utils.py:269–279

(args_data: List[ArgData])

def _test_seq_input(num_iters, operator_fn, fixed_params, input_params, input_data: ArgData, rng):
    @pipeline_def
    def pipeline(args_data: List[ArgData]):
        pos_args = [arg_data for arg_data in args_data if arg_data.desc.is_positional_arg]
        pos_nodes = [None] * len(pos_args)
        for arg_data in pos_args:
            assert 0 <= arg_data.desc.name < len(pos_nodes)
            assert pos_nodes[arg_data.desc.name] is None
            pos_nodes[arg_data.desc.name] = arg_data_node(arg_data)
        named_args = [arg_data for arg_data in args_data if not arg_data.desc.is_positional_arg]
        arg_nodes = {arg_data.desc.name: arg_data_node(arg_data) for arg_data in named_args}
        output = operator_fn(*pos_nodes, **fixed_params, **arg_nodes)
        return output

    assert num_iters >= len(input_data.data)
    max_batch_size = max(len(batch) for batch in input_data.data)