Skip to content

Commit 1c73ea4

Browse files
sumedhsakdeoclaude
andcommitted
Fix to_record_batches default order and add TaskOrder import
- Import TaskOrder from pyiceberg.table in pyarrow.py - Change to_record_batches signature to use TaskOrder() as default instead of None, ensuring consistent default scan ordering behavior Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
1 parent 84adcfa commit 1c73ea4

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

pyiceberg/io/pyarrow.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,7 @@
144144
visit,
145145
visit_with_partner,
146146
)
147-
from pyiceberg.table import DOWNCAST_NS_TIMESTAMP_TO_US_ON_WRITE, ArrivalOrder, ScanOrder, TableProperties
147+
from pyiceberg.table import DOWNCAST_NS_TIMESTAMP_TO_US_ON_WRITE, ArrivalOrder, ScanOrder, TableProperties, TaskOrder
148148
from pyiceberg.table.locations import load_location_provider
149149
from pyiceberg.table.metadata import TableMetadata
150150
from pyiceberg.table.name_mapping import NameMapping, apply_name_mapping
@@ -1837,7 +1837,7 @@ def to_table(self, tasks: Iterable[FileScanTask]) -> pa.Table:
18371837
def to_record_batches(
18381838
self,
18391839
tasks: Iterable[FileScanTask],
1840-
order: ScanOrder | None = None,
1840+
order: ScanOrder = TaskOrder(),
18411841
) -> Iterator[pa.RecordBatch]:
18421842
"""Scan the Iceberg table and return an Iterator[pa.RecordBatch].
18431843

0 commit comments

Comments
 (0)