From 5e143effc56aeb9991f86c1d5dc2ba16c2aee4c3 Mon Sep 17 00:00:00 2001 From: David del Real Date: Fri, 18 Sep 2026 18:00:52 +0000 Subject: [PATCH] build(deps): update keras and adjust e2e test configuration (#14599) * 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. * fices linting issues. * Added gemini suggestion. --- 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..6586cb48b6d 100644 --- a/dataflow/gemma/e2e_test.py +++ b/dataflow/gemma/e2e_test.py @@ -42,16 +42,18 @@ from collections.abc import Callable, Iterator +import os + import conftest # python-docs-samples/dataflow/conftest.py from conftest import Utils 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", "gs://perm-dataflow-gemma-example-testdata/gemma_2b") 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