(beam) 01/01: Cherrypick #37800

[vt...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

vterentev pushed a commit to branch cp-37800
in repository https://gitbox.apache.org/repos/asf/beam.git

commit 96e1df9f2629b627200ced093801323174361ec4
Author: Vitaly Terentyev <vi...@akvelon.com>
AuthorDate: Tue Mar 10 20:08:40 2026 +0400

    Cherrypick #37800
---
 sdks/python/setup.py | 3 +++
 sdks/python/tox.ini  | 2 +-
 2 files changed, 4 insertions(+), 1 deletion(-)

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 a86111a45c6..0f7232ace5a 100644
--- a/sdks/python/tox.ini
+++ b/sdks/python/tox.ini
@@ -110,7 +110,7 @@ deps =
   accelerate>=1.6.0
   onnx<1.19.0
 setenv =
-extras = test,gcp,dataframe,ml_test
+extras = test,gcp,dataframe,p310_ml_test
 commands =
   # Log tensorflow version for debugging
   /bin/sh -c "pip freeze | grep -E tensorflow"