Function check_iterator_build_error

dali/test/python/test_fw_iterators.py:1703–1722

(ErrorType, Iterator, glob, *args, **kwargs)

def check_iterator_build_error(ErrorType, Iterator, glob, *args, **kwargs):
    batch_size = 4
    num_gpus = 1
    pipes, _ = create_pipeline(
        lambda gpu: create_coco_pipeline(
            batch_size=batch_size,
            num_threads=4,
            shard_id=gpu,
            num_gpus=num_gpus,
            data_paths=data_sets[0],
            random_shuffle=True,
            stick_to_shard=False,
            shuffle_after_epoch=False,
            pad_last_batch=False,
        ),
        batch_size,
        num_gpus,
    )
    with assert_raises(ErrorType, glob=glob):
        Iterator(pipes, size=pipes[0].epoch_size("Reader"), *args, **kwargs)


@attr("pytorch")