def METHOD_NAME():
    vision_data = load_dataset(object_type='VisionData', use_iterable_dataset=True, n_samples=100, batch_size=50)
    batch = next(iter(vision_data))
    assert_that(batch['images'].shape, equal_to((50, 28, 28, 1)))