Run Python PostCommit vllmTests and tensorRTtests in us-west1 to avoid us-central1 T4 GPU quota exhaustion
diff --git a/sdks/python/test-suites/dataflow/common.gradle b/sdks/python/test-suites/dataflow/common.gradle index 4d07817..ece2bc5 100644 --- a/sdks/python/test-suites/dataflow/common.gradle +++ b/sdks/python/test-suites/dataflow/common.gradle
@@ -450,7 +450,9 @@ "sdk_container_image": "us.gcr.io/apache-beam-testing/python-postcommit-it/tensor_rt:latest", "sdk_location": files(configurations.distTarBall.files).singleFile, "project": "apache-beam-testing", - "region": "us-central1", + "region": "us-west1", + "staging_location": "gs://temp-storage-for-end-to-end-tests/staging-it", + "temp_location": "gs://temp-storage-for-end-to-end-tests/temp-it", "input": "gs://apache-beam-ml/testing/inputs/tensorrt_image_file_names.txt", "output": "gs://apache-beam-ml/outputs/tensorrt_predictions.txt", "engine_path": "gs://apache-beam-ml/models/ssd_mobilenet_v2_320x320_coco17_tpu-8_trt11.trt", @@ -474,7 +476,9 @@ "sdk_container_image": "us.gcr.io/apache-beam-testing/python-postcommit-it/vllm:latest", "sdk_location": files(configurations.distTarBall.files).singleFile, "project": "apache-beam-testing", - "region": "us-central1", + "region": "us-west1", + "staging_location": "gs://temp-storage-for-end-to-end-tests/staging-it", + "temp_location": "gs://temp-storage-for-end-to-end-tests/temp-it", "disk_size_gb": 75 ]