Fix pip ResolutionTooDeep

Author: aIbrahiim

.github/trigger_files/beam_PreCommit_Python_ML.json

@@ -1,4 +1,4 @@
 {
   "comment": "Modify this file in a trivial way to cause this test suite to run",
-  "revision": 3
+  "revision": 9
 }

sdks/python/apache_beam/testing/benchmarks/cloudml/requirements.txt

@@ -15,12 +15,12 @@
 #    limitations under the License.
 #
 
-dill==0.4.1
-tfx_bsl==1.16.1
-tensorflow-transform==1.16.0
-tensorflow>=2.16,<2.17
+# Core TFT dependencies with version bounds
+tfx_bsl>=1.15,<1.17
+tensorflow-transform>=1.15,<1.17
+tensorflow>=2.15,<2.16
 numpy>=1.22.0,<2.0
-tensorflow-metadata>=1.16.1,<1.17.0
+tensorflow-metadata>=1.15,<1.16
 pyarrow>=10,<11
-tensorflow-serving-api>=2.16.1,<2.20
-tf-keras>=2.16.0,<2.17
+tensorflow-serving-api>=2.15,<2.16
+tf-keras>=2.15,<2.16

sdks/python/test-suites/dataflow/common.gradle

@@ -573,13 +573,9 @@ task installTFTRequirements {
     exec {
       workingDir "$rootProject.projectDir/sdks/python/apache_beam/testing/benchmarks/cloudml/"
       executable 'sh'
-      // installGcpTest already installed apache-beam[gcp]. tensorflow-transform also
-      // lists that dependency, so a plain pip install -r can re-resolve the GCP extra
-      // and hit ResolutionTooDeep. Install TFT with --no-deps instead.
-      args '-c', ". ${envdir}/bin/activate && " +
-          "grep -v '^tensorflow-transform' requirements.txt > /tmp/cloudml_tft_base_requirements.txt && " +
-          "pip install -r /tmp/cloudml_tft_base_requirements.txt && " +
-          "pip install --no-deps tensorflow-transform==1.16.0"
+      // Use constraints file to pin versions while allowing pip to
+      // resolve compatible versions within the specified ranges in requirements.txt
+      args '-c', ". ${envdir}/bin/activate && pip install -c constraints.txt -r requirements.txt"
     }
   }
 }