Skip to content

Commit cc63d46

Browse files
committed
Fix: Address review comments on PandasBatches and docstring
- Reinstated 'Iterator[pd.DataFrame]' inheritance for 'PandasBatches' in 'bigframes/core/blocks.py'. - Removed internal type hint 'bigframes.core.blocks.PandasBatches:' from 'to_pandas_batches' docstring in 'bigframes/dataframe.py' to avoid exposing internal types in public documentation.
1 parent 1a94fe4 commit cc63d46

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

bigframes/core/blocks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@
9999

100100

101101
@dataclasses.dataclass
102-
class PandasBatches:
102+
class PandasBatches(Iterator[pd.DataFrame]):
103103
"""Interface for mutable objects with state represented by a block value object."""
104104

105105
def __init__(

bigframes/dataframe.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1925,10 +1925,9 @@ def to_pandas_batches(
19251925
over the default size limit of 10 GB.
19261926
19271927
Returns:
1928-
bigframes.core.blocks.PandasBatches:
1929-
An iterable of smaller dataframes which combine to
1930-
form the original dataframe. Results stream from bigquery,
1931-
see https://cloud.google.com/python/docs/reference/bigquery/latest/google.cloud.bigquery.table.RowIterator#google_cloud_bigquery_table_RowIterator_to_arrow_iterable
1928+
An iterable of smaller dataframes which combine to
1929+
form the original dataframe. Results stream from bigquery,
1930+
see https://cloud.google.com/python/docs/reference/bigquery/latest/google.cloud.bigquery.table.RowIterator#google_cloud_bigquery_table_RowIterator_to_arrow_iterable
19321931
"""
19331932
return self._to_pandas_batches(
19341933
page_size=page_size,

0 commit comments

Comments
 (0)