dataset dataset ¶ Reader contract: plan sample locations, then read the requested rows. Dataset ¶ Bases: ABC get_batch(batch_size: int, split: str = 'train') -> dict[str, np.ndarray] ¶ Read sequentially; job callers use the node-clocked async stream.