Pins protobuf version <= 3.20.0 to satisfy both Vizier and TensorFlow…

… requirements. PiperOrigin-RevId: 534124782
tensorflow · May 23, 2023 · 99c1395 · 99c1395
1 parent 8980ab1
commit 99c1395
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/kokoro/github/ubuntu/cpu/oldest/continuous.cfg b/kokoro/github/ubuntu/cpu/oldest/continuous.cfg
@@ -6,7 +6,7 @@ env_vars: {
 }
 env_vars: {
   key: "TF_VERSION"
-  value: "2.12.*"
+  value: "2.9.*"
 }
 env_vars: {
   key: "TAG_FILTERS"

diff --git a/kokoro/github/ubuntu/cpu/oldest/presubmit.cfg b/kokoro/github/ubuntu/cpu/oldest/presubmit.cfg
@@ -6,7 +6,7 @@ env_vars: {
 }
 env_vars: {
   key: "TF_VERSION"
-  value: "2.12.*"
+  value: "2.9.*"
 }
 env_vars: {
   key: "TAG_FILTERS"

diff --git a/setup.py b/setup.py
@@ -177,6 +177,9 @@ def get_version():
         'tensorflow>=2.9.0; platform_machine != "arm64" or platform_system != "Darwin"',
         'tensorflow-macos>=2.9.0; platform_machine == "arm64" and platform_system == "Darwin"',
         # pylint:enable=g-line-too-long
+        # TODO(b/283835852): Remove when Vizier can upgrade to protobuf >= 4.0.
+        'protobuf~=3.20.3',
+        'wrapt~=1.13.0',
     ],
     python_requires='>=3.8,<4',
     packages=find_namespace_packages(