From 3fef9973867ae8be225c1753e196006a2860cdac Mon Sep 17 00:00:00 2001
From: jrmccluskey <thejackmccluskey@gmail.com>
Date: Wed, 3 Jun 2026 15:20:01 -0400
Subject: [PATCH] [Experiment] Try single pip call + uv usage

---
 .../beam_PreCommit_Yaml_Xlang_Direct.json            |  2 +-
 .../org/apache/beam/gradle/BeamModulePlugin.groovy   | 12 +++++-------
 2 files changed, 6 insertions(+), 8 deletions(-)

diff --git a/.github/trigger_files/beam_PreCommit_Yaml_Xlang_Direct.json b/.github/trigger_files/beam_PreCommit_Yaml_Xlang_Direct.json
index 616d37428c01..8c604b0a135c 100644
--- a/.github/trigger_files/beam_PreCommit_Yaml_Xlang_Direct.json
+++ b/.github/trigger_files/beam_PreCommit_Yaml_Xlang_Direct.json
@@ -1,4 +1,4 @@
 {
     "comment": "Modify this file in a trivial way to cause this test suite to run",
-    "revision": 1
+    "revision": 2
 }
diff --git a/buildSrc/src/main/groovy/org/apache/beam/gradle/BeamModulePlugin.groovy b/buildSrc/src/main/groovy/org/apache/beam/gradle/BeamModulePlugin.groovy
index 5ca0de9de846..766c72057327 100644
--- a/buildSrc/src/main/groovy/org/apache/beam/gradle/BeamModulePlugin.groovy
+++ b/buildSrc/src/main/groovy/org/apache/beam/gradle/BeamModulePlugin.groovy
@@ -3137,15 +3137,13 @@ class BeamModulePlugin implements Plugin<Project> {
           def distTarBall = "${pythonRootDir}/build/apache-beam.tar.gz"
           def packages = "gcp,test,aws,azure,dataframe"
           def extra = project.findProperty('beamPythonExtra')
+          def installTargets = "${distTarBall}[${packages}]"
+          if (extra) {
+            installTargets = "${distTarBall}[${packages},${extra}]"
+          }
           project.exec {
             executable 'sh'
-            args '-c', ". ${project.ext.envdir}/bin/activate && pip install --pre --retries 10 ${distTarBall}[${packages}]"
-          }
-          if (extra) {
-            project.exec {
-              executable 'sh'
-              args '-c', ". ${project.ext.envdir}/bin/activate && pip install --pre --retries 10 ${distTarBall}[${extra}]"
-            }
+            args '-c', ". ${project.ext.envdir}/bin/activate && pip install uv && uv pip install --pre ${installTargets}"
           }
         }
       }