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 b1676adfd4648f6e5794ec9c04b26abb3bc56048 Author: Derrick Williams <[email protected]> AuthorDate: Sun May 24 18:20:12 2026 +0000 Fix installYamlIntegrationTestDeps to use constraint files and prevent pip backtracking hangs --- sdks/python/build.gradle | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/sdks/python/build.gradle b/sdks/python/build.gradle index 837631868b8..0ae4aaa4148 100644 --- a/sdks/python/build.gradle +++ b/sdks/python/build.gradle @@ -133,8 +133,19 @@ tasks.register("installYamlIntegrationTestDeps") { "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 && " + - "echo \"Installing dependencies...\" && " + - "pip install --pre --retries 10 ${buildDir}/apache-beam.tar.gz[\$ml_extra,yaml,transformers]" + "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_extra,yaml,transformers]; " + + "else " + + " echo \"No constraint file found, installing without constraints\" && " + + " pip install --pre --retries 10 \${buildDir}/apache-beam.tar.gz[\$ml_extra,yaml,transformers]; " + + "fi" } } }
