Skip to content

Commit cf536ea

Browse files
authored
Fix 36181 cloudml benchmarks job (#37803)
* Pin cloudml benchmark deps to avoid pip resolution-too-deep on Dataflow * Reduce Dataflow inactivity timeout risk for TFT CloudML benchmark * Tighten CloudML TFT benchmark requirements * focus fix on dependency bounds only
1 parent 9d73d76 commit cf536ea

2 files changed

Lines changed: 10 additions & 4 deletions

File tree

.github/workflows/load-tests-pipeline-options/beam_CloudML_Benchmarks_Dataflow_arguments.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,4 +20,4 @@
2020
--staging_location=gs://temp-storage-for-perf-tests/loadtests
2121
--temp_location=gs://temp-storage-for-perf-tests/loadtests
2222
--runner=DataflowRunner
23-
--requirements_file=apache_beam/testing/benchmarks/cloudml/requirements.txt
23+
--requirements_file=apache_beam/testing/benchmarks/cloudml/requirements.txt

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

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,12 @@
1515
# limitations under the License.
1616
#
1717

18-
dill
19-
tfx_bsl
20-
tensorflow-transform
18+
dill==0.4.1
19+
tfx_bsl==1.16.1
20+
tensorflow-transform==1.16.0
21+
tensorflow>=2.16,<2.17
22+
numpy>=1.22.0,<2.0
23+
tensorflow-metadata>=1.16.1,<1.17.0
24+
pyarrow>=10,<11
25+
tensorflow-serving-api>=2.16.1,<2.20
26+
tf-keras>=2.16.0,<2.17

0 commit comments

Comments
 (0)