Skip to content

Commit 74c7154

Browse files
Fix batch_size default description in to_polars docstrings (#7824)
1 parent 0b2a4c2 commit 74c7154

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/datasets/arrow_dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5189,7 +5189,7 @@ def to_polars(
51895189
Args:
51905190
batch_size (`int`, *optional*):
51915191
The size (number of rows) of the batches if `batched` is `True`.
5192-
Defaults to `genomicsml.datasets.config.DEFAULT_MAX_BATCH_SIZE`.
5192+
Defaults to `datasets.config.DEFAULT_MAX_BATCH_SIZE`.
51935193
batched (`bool`):
51945194
Set to `True` to return a generator that yields the dataset as batches
51955195
of `batch_size` rows. Defaults to `False` (returns the whole datasets once).

src/datasets/iterable_dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3671,7 +3671,7 @@ def to_polars(
36713671
Args:
36723672
batch_size (`int`, *optional*):
36733673
The size (number of rows) of the batches if `batched` is `True`.
3674-
Defaults to `genomicsml.datasets.config.DEFAULT_MAX_BATCH_SIZE`.
3674+
Defaults to `datasets.config.DEFAULT_MAX_BATCH_SIZE`.
36753675
batched (`bool`):
36763676
Set to `True` to return a generator that yields the dataset as batches
36773677
of `batch_size` rows. Defaults to `False` (returns the whole datasets once).

0 commit comments

Comments
 (0)