Skip to content

Increment PyArrow Bounds #37503

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
jrmccluskey merged 3 commits into from
Feb 4, 2026
Merged

Increment PyArrow Bounds #37503

Show file tree
Hide file tree
Changes from 2 commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
7268661
Increment Pyarrow Bounds
jrmccluskey Feb 4, 2026
ba8b542
fix dynamic work rebalancing test
jrmccluskey Feb 4, 2026
1d93aa9
fix pandas pin
jrmccluskey Feb 4, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion sdks/python/apache_beam/io/parquetio_test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -564,7 +564,7 @@ def test_dynamic_work_rebalancing(self):
# count to row_group_size also being sufficiently large (but the required
# ratio to pass varies for values of row_group_size and, somehow, the
# version of pyarrow being tested against.)
file_name = self._write_data(count=280, row_group_size=20)
file_name = self._write_data(count=320, row_group_size=20)
source = _create_parquet_source(file_name)

splits = [split for split in source.split(desired_bundle_size=float('inf'))]
Expand Down
2 changes: 1 addition & 1 deletion sdks/python/setup.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -145,7 +145,7 @@ def cythonize(*args, **kwargs):
pyarrow_dependency = ['']
else:
pyarrow_dependency = [
'pyarrow>=3.0.0,<19.0.0',
'pyarrow>=6.0.1,<24.0.0',
# NOTE(https://github.com/apache/beam/issues/29392): We can remove this
# once Beam increases the pyarrow lower bound to a version that fixes CVE.
# (lower bound >= 14.0.1)
Expand Down
18 changes: 7 additions & 11 deletions sdks/python/tox.ini
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -343,26 +343,22 @@ extras = test
commands =
bash {toxinidir}/scripts/pytest_validates_runner.sh {envname} {toxinidir}/apache_beam/runners/portability/prism_runner_test.py {posargs}

[testenv:py{310,311}-pyarrow-{3,9,10,11,12,13,14,15,16,17,18}]
[testenv:py{310,311}-pyarrow-{6,15,16,17,18,19,20,21,22,23}]
deps =
# As a courtesy to users, test against the oldest allowed version of Pyarrow.
# We'd have to increase the pyarrow lower bound when Python 3.9 is deprecated.
# Since Pandas 2 requires pyarrow>=7, downgrade pandas for this test.
3: pyarrow>=3,<4
3: pandas<2
3: numpy>=1.14.3,<1.27.0
6: pyarrow>=6,<7
# Test against versions of pyarrow released in last ~2 years.
9: pyarrow>=9,<10
9: pandas==2.1.4
10: pyarrow>=10,<11
11: pyarrow>=11,<12
12: pyarrow>=12,<13
13: pyarrow>=13,<14
14: pyarrow>=14,<15
15: pyarrow>=15,<16
16: pyarrow>=16,<17
17: pyarrow>=17,<18
18: pyarrow>=18,<19
19: pyarrow>=19,<20
20: pyarrow>=20,<21
21: pyarrow>=21,<22
22: pyarrow>=22,<23
23: pyarrow>=23,<24
numpy==1.26.4
commands =
# Log pyarrow and numpy version for debugging
Expand Down
Loading