From f75cd0d4d46b97fd909e0cedebba8bfde1f6181d Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Fri, 27 Feb 2026 12:07:32 +0800 Subject: [PATCH] fix(CI): pin protobuf<7.34 See https://github.com/tensorflow/tensorflow/commit/23f7b26bc54ec47e63c0d6e2cde12e1fa653f960 Signed-off-by: Jinzhe Zeng --- pyproject.toml | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 86a7e569b2..9575d237cc 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -153,12 +153,18 @@ dev = [ pin_tensorflow_cpu = [ # https://github.com/tensorflow/tensorflow/issues/75279 # macos x86 has been deprecated - "tensorflow-cpu>=2.18,<2.21; platform_machine=='x86_64' and platform_system == 'Linux'", + "tensorflow-cpu==2.20.0; platform_machine=='x86_64' and platform_system == 'Linux'", "tensorflow~=2.18.0; (platform_machine!='x86_64' or platform_system != 'Linux') and (platform_machine!='x86_64' or platform_system != 'Darwin')", "tensorflow; platform_machine=='x86_64' and platform_system == 'Darwin'", + # TODO: unpin protobuf after TF is upgraded to 2.21 + # See: https://github.com/tensorflow/tensorflow/pull/103382 + "protobuf<7.34.0", ] pin_tensorflow_gpu = [ - "tensorflow~=2.18.0", + "tensorflow==2.18.0", + # TODO: unpin protobuf after TF is upgraded to 2.21 + # See: https://github.com/tensorflow/tensorflow/pull/103382 + "protobuf<7.34.0", ] pin_pytorch_cpu = [ # https://github.com/pytorch/pytorch/issues/114602