build: [MLG-336] Limit the version of protobuf

Installing the requirement `tensorflow-macos=2.8.0` pulls protobuf as a downstream dependency. Version 4.21 of the Python protobuf package had a breaking change that makes it incompatible with tensorflow-2.8.0 (see tensorflow/tensorflow#56077). Later patches to Tensorflow limit the version of protobuf to 3.20. We've got a work item to update the tensorflow we include, but until then this change gives the ceiling on tensorflow's protobuf dependency that its later versions enforce.
wes-turner · Mar 2, 2023 · 1922bca · 1922bca
1 parent eac86af
commit 1922bca
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/e2e_tests/tests/requirements.txt b/e2e_tests/tests/requirements.txt
@@ -1,4 +1,5 @@
 appdirs
+protobuf<3.20  # TODO(MLG-336): remove this requirement when tensorflow-macos is upgraded.
 # pytest 6.0 has linter-breaking changes
 pytest>=6.0.1
 pytest-timeout

diff --git a/examples/tests/requirements.txt b/examples/tests/requirements.txt
@@ -1,4 +1,5 @@
 # pytest 6.0 has linter-breaking changes
+protobuf<3.20  # TODO(MLG-336): remove this requirement when tensorflow-macos is upgraded.
 pytest>=6.0.1
 tensorflow==2.8.4; sys_platform != 'darwin' or platform_machine != 'arm64'
 tensorflow-macos==2.8.0; sys_platform == 'darwin' and platform_machine == 'arm64'