Function gen_batch

dali/test/python/conditionals/test_pipeline_conditionals.py:154–155

(generator, bs, iter)

    pipeline_definition = pipeline_def(enable_conditionals=True)(function)

    def gen_batch(generator, bs, iter):
        return [generator(SampleInfo(bs * iter + i, i, iter, 0)) for i in range(bs)]

    pipe = pipeline_definition(*es_inputs, **kwargs)