(beam) 03/04: update dependency logic

[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

commit c5181114e0825ee908aeb82499187af308f42979
Author: Derrick Williams <derric...@google.com>
AuthorDate: Tue May 12 02:45:51 2026 +0000

    update dependency logic
---
 sdks/python/build.gradle | 17 ++++-------------
 1 file changed, 4 insertions(+), 13 deletions(-)

diff --git a/sdks/python/build.gradle b/sdks/python/build.gradle
index 8450e32d593..837631868b8 100644
--- a/sdks/python/build.gradle
+++ b/sdks/python/build.gradle
@@ -131,19 +131,10 @@ 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}\")') && " +
-              "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\"; " +
-              "elif [ -f 
\"container/py\${py_ver}/base_image_requirements.txt\" ]; then " +
-              "  
constraint_file=\"container/py\${py_ver}/base_image_requirements.txt\"; " +
-              "fi && " +
-              "if [ -n \"\$constraint_file\" ]; then " +
-              "  echo \"Installing with constraint file: \$constraint_file\" 
&& " +
-              "  pip install --pre --retries 10 -c \"\$constraint_file\" 
\${buildDir}/apache-beam.tar.gz[ml_test,yaml,transformers]; " +
-              "else " +
-              "  echo \"No constraint file found, installing without 
constraints\" && " +
-              "  pip install --pre --retries 10 
\${buildDir}/apache-beam.tar.gz[ml_test,yaml,transformers]; " +
-              "fi"
+              "ml_extra=\"ml_test\" && " +
+              "if [ \"\$py_ver\" -ge 313 ]; then 
ml_extra=\"p\${py_ver}_ml_test\"; fi && " +
+              "echo \"Installing dependencies...\" && " +
+              "pip install --pre --retries 10 
${buildDir}/apache-beam.tar.gz[\$ml_extra,yaml,transformers]"
     }
   }
 }