chore(gemma2/dataflow): update google-cloud-aiplatform and protobuf for gemma2 (#14350)

XrossFox · web-flow · commit a5dbe887832f · 2026-06-24T20:24:56.000-07:00
- Update `google-cloud-aiplatform` to 1.158.0 in `gemma2/` and `dataflow/` samples
 - Update `protobuf` to 6.33.6 in `gemma2/`
 - Remove python version constraint for pytest in `gemma2/`
diff --git a/dataflow/gemma-flex-template/requirements-test.txt b/dataflow/gemma-flex-template/requirements-test.txt
@@ -1,4 +1,4 @@
-google-cloud-aiplatform==1.62.0
+google-cloud-aiplatform==1.158.0
 google-cloud-dataflow-client==0.8.14
 google-cloud-pubsub==2.28.0
 google-cloud-storage==2.18.2
diff --git a/dataflow/gemma/requirements-test.txt b/dataflow/gemma/requirements-test.txt
@@ -1,4 +1,4 @@
-google-cloud-aiplatform==1.49.0
+google-cloud-aiplatform==1.158.0
 google-cloud-dataflow-client==0.8.10
 google-cloud-storage==2.16.0
 pytest==9.0.3; python_version >= "3.10"
diff --git a/dataflow/run-inference/requirements-test.txt b/dataflow/run-inference/requirements-test.txt
@@ -1,4 +1,4 @@
-google-cloud-aiplatform==1.57.0
+google-cloud-aiplatform==1.158.0
 google-cloud-dataflow-client==0.8.14
 google-cloud-storage==2.10.0
 pytest==9.0.3; python_version >= "3.10"
diff --git a/gemma2/requirements-test.txt b/gemma2/requirements-test.txt
@@ -1 +1 @@
-pytest==9.0.3; python_version >= "3.10"
+pytest==9.0.3
diff --git a/gemma2/requirements.txt b/gemma2/requirements.txt
@@ -1,2 +1,2 @@
-google-cloud-aiplatform[full]==1.157.0
-protobuf==5.29.5
+google-cloud-aiplatform==1.158.0
+protobuf==6.33.6

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-google-cloud-aiplatform==1.62.0`
	`1`	`+google-cloud-aiplatform==1.158.0`
`2`	`2`	`google-cloud-dataflow-client==0.8.14`
`3`	`3`	`google-cloud-pubsub==2.28.0`
`4`	`4`	`google-cloud-storage==2.18.2`
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-google-cloud-aiplatform==1.49.0`
	`1`	`+google-cloud-aiplatform==1.158.0`
`2`	`2`	`google-cloud-dataflow-client==0.8.10`
`3`	`3`	`google-cloud-storage==2.16.0`
`4`	`4`	`pytest==9.0.3; python_version >= "3.10"`
Original file line number	Diff line number	Diff line change
`@@ -1 +1 @@`
`1`		`-pytest==9.0.3; python_version >= "3.10"`
	`1`	`+pytest==9.0.3`