From 52cc33927d04a8365162a03b1395ae9a1f8dda70 Mon Sep 17 00:00:00 2001
From: David del Real Sifuentes <deldavid@google.com>
Date: Wed, 9 Sep 2026 23:17:25 +0000
Subject: [PATCH 1/3]  build(deps): update keras and adjust e2e test
 configuration

 Bumps `keras` to version 3.15.1 and updates `pyOpenSSL` in the Dataflow gemma requirements to resolve Dependabot security alerts. Also updates `e2e_test.py` to read the `GEMMA_GCS` bucket path
 from the environment.
---
 dataflow/gemma/e2e_test.py      | 8 +++++---
 dataflow/gemma/requirements.txt | 5 ++---
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/dataflow/gemma/e2e_test.py b/dataflow/gemma/e2e_test.py
index fdb187d4b91..3762a1186b0 100644
--- a/dataflow/gemma/e2e_test.py
+++ b/dataflow/gemma/e2e_test.py
@@ -40,6 +40,8 @@
       add the PYTHONPATH to the "env" in your noxfile_config.py file.
 """
 
+import os
+
 from collections.abc import Callable, Iterator
 
 import conftest  # python-docs-samples/dataflow/conftest.py
@@ -48,10 +50,10 @@
 import pytest
 
 DATAFLOW_MACHINE_TYPE = "g2-standard-8"
-# TODO(developer): For local testing, ensure the 'gemma_2b_en' directory is uploaded 
-# to a GCS bucket you manage. Update the constant below to point to 
+# TODO(developer): For local testing, ensure the 'gemma_2b_en' directory is uploaded
+# to a GCS bucket you manage. Update the constant below to point to
 # the root path of this uploaded directory (e.g., 'gs://your-bucket-name/path/to/gemma_2b').
-GEMMA_GCS = "gs://perm-dataflow-gemma-example-testdata/gemma_2b"
+GEMMA_GCS = os.getenv("GEMMA_GCS")
 NAME = "dataflow/gemma/streaming"
 
 
diff --git a/dataflow/gemma/requirements.txt b/dataflow/gemma/requirements.txt
index 183b6016d7d..0a1798ba4da 100644
--- a/dataflow/gemma/requirements.txt
+++ b/dataflow/gemma/requirements.txt
@@ -1,5 +1,4 @@
-protobuf==6.33.6
 apache_beam[gcp]==2.74.0
-keras==3.14.1
+keras==3.15.1
 keras_nlp==0.29.1
-pyOpenSSL==26.3.0
+pyOpenSSL==26.2.0

From 91d783fa4e16ea1d3bcd0e425a8301754353ea5d Mon Sep 17 00:00:00 2001
From: David del Real Sifuentes <deldavid@google.com>
Date: Wed, 9 Sep 2026 23:25:54 +0000
Subject: [PATCH 2/3] fices linting issues.

---
 dataflow/gemma/e2e_test.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/dataflow/gemma/e2e_test.py b/dataflow/gemma/e2e_test.py
index 3762a1186b0..517d7326e74 100644
--- a/dataflow/gemma/e2e_test.py
+++ b/dataflow/gemma/e2e_test.py
@@ -40,10 +40,10 @@
       add the PYTHONPATH to the "env" in your noxfile_config.py file.
 """
 
-import os
-
 from collections.abc import Callable, Iterator
 
+import os
+
 import conftest  # python-docs-samples/dataflow/conftest.py
 from conftest import Utils
 

From 78f72aa6351555fc020d6af9473798089285052d Mon Sep 17 00:00:00 2001
From: David del Real Sifuentes <deldavid@google.com>
Date: Thu, 10 Sep 2026 22:21:47 +0000
Subject: [PATCH 3/3] Added gemini suggestion.

---
 dataflow/gemma/e2e_test.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dataflow/gemma/e2e_test.py b/dataflow/gemma/e2e_test.py
index 517d7326e74..6586cb48b6d 100644
--- a/dataflow/gemma/e2e_test.py
+++ b/dataflow/gemma/e2e_test.py
@@ -53,7 +53,7 @@
 # TODO(developer): For local testing, ensure the 'gemma_2b_en' directory is uploaded
 # to a GCS bucket you manage. Update the constant below to point to
 # the root path of this uploaded directory (e.g., 'gs://your-bucket-name/path/to/gemma_2b').
-GEMMA_GCS = os.getenv("GEMMA_GCS")
+GEMMA_GCS = os.getenv("GEMMA_GCS", "gs://perm-dataflow-gemma-example-testdata/gemma_2b")
 NAME = "dataflow/gemma/streaming"