Function expected

python/ray/data/tests/test_random_access.py:20–21

(i)

    rad = ds.to_random_access_dataset("key", num_workers=1)

    def expected(i):
        return {"id": i, "key": i * 2, "embedding": i**2}

    # Test get.
    assert ray.get(rad.get_async(-1)) is None