Function test_paddle_prepare_first_batch

dali/test/python/test_fw_iterators.py:1662–1667

()

@attr("paddle")
def test_paddle_prepare_first_batch():
    from nvidia.dali.plugin.paddle import DALIGenericIterator as PaddleIterator

    check_prepare_first_batch(
        PaddleIterator, output_map=["data"], to_np=lambda x: np.array(x["data"])
    )


@attr("jax")