diff --git a/src/datasets/table.py b/src/datasets/table.py index 376e1e960f5..b5e039faabf 100644 --- a/src/datasets/table.py +++ b/src/datasets/table.py @@ -2152,6 +2152,8 @@ def _visit(array, feature): def table_iter(pa_table: pa.Table, batch_size: int, drop_last_batch=False): """Iterate ober sub-tables of size `batch_size`. + + Requires pyarrow>=8.0.0 Args: table (:obj:`pyarrow.Table`): PyArrow table to iterate over