ray.data.Dataset.take
ray.data.Dataset.take#
- Dataset.take(limit: int = 20) List[ray.data.block.T] [source]#
Return up to
limit
records from the dataset.This will move up to
limit
records to the caller’s machine; iflimit
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.