From 43d287b0fa16e0e72851ebf4ca127f66a37b0ab7 Mon Sep 17 00:00:00 2001 From: David del Real Sifuentes Date: Thu, 24 Sep 2026 22:45:51 +0000 Subject: [PATCH] chore(deps): update transformers and dependencies in dataflow/run-inference Bumps `transformers` (to 5.17.0), `apache-beam[gcp]` (to 2.76.0), `torch` (to 2.12.1), and updates various test dependencies (`google-cloud-aiplatform`, `google-cloud-dataflow-client`, `google-cloud-storage`, `pytest`) to resolve Dependabot security alerts. Also adjusts Python testing versions in `noxfile_config.py`. Note on test failure and deprecation: One test is currently failing due to obsolete usage of pre-built containers which only support up to `torch 2.4`. Rather than repairing this, the sample has been deemed obsolete and is no longer feasible to maintain. A new page in the documentation supersedes this sample: https://docs.cloud.google.com/dataflow/docs/notebooks/run_inference_pytorch The old doc (https://docs.cloud.google.com/dataflow/docs/tutorials/streaming-llm.md) and this sample should both be retired. This deprecation and removal process is being tracked in b/565898242. --- dataflow/run-inference/noxfile_config.py | 2 +- dataflow/run-inference/requirements-test.txt | 8 ++++---- dataflow/run-inference/requirements.txt | 4 ++-- 3 files changed, 7 insertions(+), 7 deletions(-) diff --git a/dataflow/run-inference/noxfile_config.py b/dataflow/run-inference/noxfile_config.py index feeabc69601..203d2e941a6 100644 --- a/dataflow/run-inference/noxfile_config.py +++ b/dataflow/run-inference/noxfile_config.py @@ -22,7 +22,7 @@ TEST_CONFIG_OVERRIDE = { # You can opt out from the test for specific Python versions. - "ignored_versions": ["3.8", "3.9", "3.11", "3.12", "3.13"], + "ignored_versions": ["3.8", "3.9", "3.11", "3.12", "3.14"], # Old samples are opted out of enforcing Python type hints # All new samples should feature them "enforce_type_hints": True, diff --git a/dataflow/run-inference/requirements-test.txt b/dataflow/run-inference/requirements-test.txt index a88ea434900..e5d8ab77526 100644 --- a/dataflow/run-inference/requirements-test.txt +++ b/dataflow/run-inference/requirements-test.txt @@ -1,4 +1,4 @@ -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" +google-cloud-aiplatform==2.2.0 +google-cloud-dataflow-client==0.14.0 +google-cloud-storage==3.14.1 +pytest==9.1.1 diff --git a/dataflow/run-inference/requirements.txt b/dataflow/run-inference/requirements.txt index 9b2b60247c7..9b82cfa92f2 100644 --- a/dataflow/run-inference/requirements.txt +++ b/dataflow/run-inference/requirements.txt @@ -1,3 +1,3 @@ -apache-beam[gcp]==2.49.0 +apache-beam[gcp]==2.76.0 torch==2.12.1 -transformers==5.0.0rc3 +transformers==5.17.0