Function get_data_source

dali/test/python/test_dali_cpu_only.py:1128–1129

(*args, **kwargs)

def test_segmentation_select_masks():
    def get_data_source(*args, **kwargs):
        return lambda: make_batch_select_masks(*args, **kwargs)

    pipe = Pipeline(batch_size=batch_size, num_threads=4, device_id=None, seed=1234)
    with pipe: