This is an automated email from the ASF dual-hosted git repository.
vterentev pushed a commit to branch fix-publish-sdk
in repository https://gitbox.apache.org/repos/asf/beam.git
The following commit(s) were added to refs/heads/fix-publish-sdk by this push:
new 24b565eff27 Fix tensorflow and triton versions
24b565eff27 is described below
commit 24b565eff276efa36998b3d74f1b7a224672e3e8
Author: Vitaly Terentyev <[email protected]>
AuthorDate: Tue Oct 21 16:20:08 2025 +0400
Fix tensorflow and triton versions
---
sdks/python/container/ml/py310/base_image_requirements.txt | 4 ++--
sdks/python/container/ml/py310/ml_image_requirements.txt | 2 +-
sdks/python/container/ml/py311/base_image_requirements.txt | 4 ++--
sdks/python/container/ml/py311/ml_image_requirements.txt | 2 +-
sdks/python/container/ml/py312/base_image_requirements.txt | 4 ++--
sdks/python/container/ml/py312/ml_image_requirements.txt | 2 +-
sdks/python/container/ml/py313/ml_image_requirements.txt | 2 +-
sdks/python/container/ml/py39/base_image_requirements.txt | 2 +-
8 files changed, 11 insertions(+), 11 deletions(-)
diff --git a/sdks/python/container/ml/py310/base_image_requirements.txt
b/sdks/python/container/ml/py310/base_image_requirements.txt
index a7c789801ea..53a1eb1c874 100644
--- a/sdks/python/container/ml/py310/base_image_requirements.txt
+++ b/sdks/python/container/ml/py310/base_image_requirements.txt
@@ -219,7 +219,7 @@ sympy==1.14.0
tenacity==8.5.0
tensorboard==2.16.2
tensorboard-data-server==0.7.2
-tensorflow==2.16.2
+tensorflow==2.20.0
tensorflow-cpu-aws==2.15.1;platform_machine=="aarch64"
tensorflow-io-gcs-filesystem==0.37.1
termcolor==3.1.0
@@ -230,7 +230,7 @@ tomli==2.2.1
torch==2.7.1
tqdm==4.67.1
transformers==4.55.4
-triton==3.3.1
+triton==3.5.0
typing-inspection==0.4.1
typing_extensions==4.14.1
tzdata==2025.2
diff --git a/sdks/python/container/ml/py310/ml_image_requirements.txt
b/sdks/python/container/ml/py310/ml_image_requirements.txt
index b30a5d650fa..d7597d4a051 100644
--- a/sdks/python/container/ml/py310/ml_image_requirements.txt
+++ b/sdks/python/container/ml/py310/ml_image_requirements.txt
@@ -231,7 +231,7 @@ tomli==2.3.0
torch==2.7.1
tqdm==4.67.1
transformers==4.55.4
-triton==3.3.1
+triton==3.5.0
typing-inspection==0.4.2
typing_extensions==4.15.0
tzdata==2025.2
diff --git a/sdks/python/container/ml/py311/base_image_requirements.txt
b/sdks/python/container/ml/py311/base_image_requirements.txt
index b9259dd102d..25574896706 100644
--- a/sdks/python/container/ml/py311/base_image_requirements.txt
+++ b/sdks/python/container/ml/py311/base_image_requirements.txt
@@ -217,7 +217,7 @@ sympy==1.14.0
tenacity==8.5.0
tensorboard==2.16.2
tensorboard-data-server==0.7.2
-tensorflow==2.16.2
+tensorflow==2.20.0
tensorflow-cpu-aws==2.15.1;platform_machine=="aarch64"
tensorflow-io-gcs-filesystem==0.37.1
termcolor==3.1.0
@@ -227,7 +227,7 @@ tokenizers==0.21.4
torch==2.7.1
tqdm==4.67.1
transformers==4.55.4
-triton==3.3.1
+triton==3.5.0
typing-inspection==0.4.1
typing_extensions==4.14.1
tzdata==2025.2
diff --git a/sdks/python/container/ml/py311/ml_image_requirements.txt
b/sdks/python/container/ml/py311/ml_image_requirements.txt
index deb06bf588e..7a73aa95187 100644
--- a/sdks/python/container/ml/py311/ml_image_requirements.txt
+++ b/sdks/python/container/ml/py311/ml_image_requirements.txt
@@ -228,7 +228,7 @@ tokenizers==0.21.4
torch==2.7.1
tqdm==4.67.1
transformers==4.55.4
-triton==3.3.1
+triton==3.5.0
typing-inspection==0.4.2
typing_extensions==4.15.0
tzdata==2025.2
diff --git a/sdks/python/container/ml/py312/base_image_requirements.txt
b/sdks/python/container/ml/py312/base_image_requirements.txt
index f0f94f12b51..1eed3ab7862 100644
--- a/sdks/python/container/ml/py312/base_image_requirements.txt
+++ b/sdks/python/container/ml/py312/base_image_requirements.txt
@@ -216,7 +216,7 @@ sympy==1.14.0
tenacity==8.5.0
tensorboard==2.16.2
tensorboard-data-server==0.7.2
-tensorflow==2.16.2
+tensorflow==2.20.0
tensorflow-cpu-aws==2.15.1;platform_machine=="aarch64"
termcolor==3.1.0
testcontainers==4.12.0
@@ -225,7 +225,7 @@ tokenizers==0.21.4
torch==2.7.1
tqdm==4.67.1
transformers==4.55.4
-triton==3.3.1
+triton==3.5.0
typing-inspection==0.4.1
typing_extensions==4.14.1
tzdata==2025.2
diff --git a/sdks/python/container/ml/py312/ml_image_requirements.txt
b/sdks/python/container/ml/py312/ml_image_requirements.txt
index 97ed9671ec1..32de9dbb61d 100644
--- a/sdks/python/container/ml/py312/ml_image_requirements.txt
+++ b/sdks/python/container/ml/py312/ml_image_requirements.txt
@@ -227,7 +227,7 @@ tokenizers==0.21.4
torch==2.7.1
tqdm==4.67.1
transformers==4.55.4
-triton==3.3.1
+triton==3.5.0
typing-inspection==0.4.2
typing_extensions==4.15.0
tzdata==2025.2
diff --git a/sdks/python/container/ml/py313/ml_image_requirements.txt
b/sdks/python/container/ml/py313/ml_image_requirements.txt
index 969541673ab..80c1f6fca6f 100644
--- a/sdks/python/container/ml/py313/ml_image_requirements.txt
+++ b/sdks/python/container/ml/py313/ml_image_requirements.txt
@@ -224,7 +224,7 @@ tokenizers==0.21.4
torch==2.7.1
tqdm==4.67.1
transformers==4.55.4
-triton==3.3.1
+triton==3.5.0
typing-inspection==0.4.2
typing_extensions==4.15.0
tzdata==2025.2
diff --git a/sdks/python/container/ml/py39/base_image_requirements.txt
b/sdks/python/container/ml/py39/base_image_requirements.txt
index e4009c9071e..1bb117a8cb8 100644
--- a/sdks/python/container/ml/py39/base_image_requirements.txt
+++ b/sdks/python/container/ml/py39/base_image_requirements.txt
@@ -219,7 +219,7 @@ sympy==1.14.0
tenacity==8.5.0
tensorboard==2.16.2
tensorboard-data-server==0.7.2
-tensorflow==2.16.2
+tensorflow==2.20.0
tensorflow-cpu-aws==2.15.1;platform_machine=="aarch64"
tensorflow-io-gcs-filesystem==0.37.1
termcolor==3.1.0
