(beam) 01/01: Use p310_ml_test

[vt...@apache.org]

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

commit 78b22bc11f2c445f8977fea2bf14d9d7113672ba
Author: Vitaly Terentyev <vi...@akvelon.com>
AuthorDate: Mon Mar 9 19:49:58 2026 +0400

    Use p310_ml_test
---
 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 71b9b27cfcf..cb84425a35a 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"