ray.data.Dataset.take

Dataset.take(limit: int = 20) → List[ray.data.block.T]

Return up to limit records from the dataset.

This will move up to limit records to the caller’s machine; if limit is very large, this can result in an OutOfMemory crash on the caller.

Time complexity: O(limit specified)

Parameters

limit – The max number of records to return.

Returns

A list of up to limit records from the dataset.