This is an automated email from the ASF dual-hosted git repository.
vterentev pushed a commit to branch fix-python-ml-3-10
in repository https://gitbox.apache.org/repos/asf/beam.git
The following commit(s) were added to refs/heads/fix-python-ml-3-10 by this
push:
new dfbe47a78cc Use p310_ml_test
dfbe47a78cc is described below
commit dfbe47a78cc0b1e139db37738feaecb72383c92c
Author: Vitaly Terentyev <[email protected]>
AuthorDate: Mon Mar 9 23:28:29 2026 +0400
Use p310_ml_test
---
sdks/python/setup.py | 3 +++
sdks/python/tox.ini | 4 ++--
2 files changed, 5 insertions(+), 2 deletions(-)
diff --git a/sdks/python/setup.py b/sdks/python/setup.py
index e45bb55ec3e..81e8acb82de 100644
--- a/sdks/python/setup.py
+++ b/sdks/python/setup.py
@@ -550,6 +550,9 @@ if __name__ == '__main__':
# https://github.com/apache/beam/issues/31285
# 'xgboost<2.0', # https://github.com/apache/beam/issues/31252
] + ml_base,
+ 'p310_ml_test': [
+ 'datatable',
+ ] + ml_base,
'p312_ml_test': [
'datatable',
] + ml_base,
diff --git a/sdks/python/tox.ini b/sdks/python/tox.ini
index 857bc5ba85c..2c05b82f46d 100644
--- a/sdks/python/tox.ini
+++ b/sdks/python/tox.ini
@@ -106,11 +106,11 @@ commands =
[testenv:py{310,311}-ml]
# Don't set TMPDIR to avoid "AF_UNIX path too long" errors in certain tests.
deps =
- pip==26.0.1
+ pip==25.0.1
accelerate>=1.6.0
onnx<1.19.0
setenv =
-extras = test,gcp,dataframe,ml_test
+extras = test,gcp,dataframe,p310_ml_test
pip_pre = False
commands =
# Log tensorflow version for debugging
