Merge pull request #37800 from apache/fix-python-ml-3-10

Fix PreCommit Python ML tests with ML deps installed

Author: Amar3tto

sdks/python/setup.py

@@ -550,6 +550,9 @@ def get_portability_package_data():
               # 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,

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"