(beam) branch 20260524_huggingFaceTry#3 updated: fix ml_test dependencies

[derrick]

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

derrickaw pushed a commit to branch 20260524_huggingFaceTry#3
in repository https://gitbox.apache.org/repos/asf/beam.git


The following commit(s) were added to refs/heads/20260524_huggingFaceTry#3 by 
this push:
     new ebf9aac5408 fix ml_test dependencies
ebf9aac5408 is described below

commit ebf9aac5408c142ef1c3209606f158d52dd039d1
Author: Derrick Williams <derric...@google.com>
AuthorDate: Tue May 26 15:29:52 2026 +0000

    fix ml_test dependencies
---
 sdks/python/build.gradle | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/sdks/python/build.gradle b/sdks/python/build.gradle
index d9c5aff82b6..cfe4e87c9c4 100644
--- a/sdks/python/build.gradle
+++ b/sdks/python/build.gradle
@@ -131,8 +131,9 @@ tasks.register("installYamlIntegrationTestDeps") {
       executable 'sh'
       args '-c', ". ${envdir}/bin/activate && " +
               "py_ver=\$(python -c 'import sys; 
print(f\"{sys.version_info.major}{sys.version_info.minor}\")') && " +
-              "ml_extra=\"ml_test\" && " +
-              "if [ \"\$py_ver\" -ge 313 ]; then 
ml_extra=\"p\${py_ver}_ml_test\"; fi && " +
+              "ml_extra=\"p310_ml_test\" && " +
+              "if [ \"\$py_ver\" -eq 312 ]; then ml_extra=\"p312_ml_test\"; " +
+              "elif [ \"\$py_ver\" -ge 313 ]; then 
ml_extra=\"p\${py_ver}_ml_test\"; fi && " +
               "constraint_file=\"\" && " +
               "if [ -f 
\"container/ml/py\${py_ver}/base_image_requirements.txt\" ]; then " +
               "  
constraint_file=\"container/ml/py\${py_ver}/base_image_requirements.txt\"; " +