Use p310_ml_test

sdks/python/setup.py

diff --git a/sdks/python/setup.py b/sdks/python/setup.py
index e45bb55..81e8acb 100644
--- a/sdks/python/setup.py
+++ b/sdks/python/setup.py
@@ -550,6 +550,9 @@
               # 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

diff --git a/sdks/python/tox.ini b/sdks/python/tox.ini
index 71b9b27..cb84425 100644
--- a/sdks/python/tox.ini
+++ b/sdks/python/tox.ini
@@ -110,7 +110,7 @@
   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"