limit the protobuf version - tensorflow requirement (openvinotoolkit#…

…1199)
vshampor · Jun 21, 2022 · e16bed2 · e16bed2
1 parent d351ad6
commit e16bed2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -135,6 +135,9 @@ def find_version(*file_paths):
     sys.argv.remove("--torch")
 
 if "--tf" in sys.argv:
+    # See also: https://github.com/tensorflow/tensorflow/issues/56077
+    # This is a temporary patch, that limites the protobuf version from above
+    INSTALL_REQUIRES.extend(["protobuf>=3.9.2,<3.20"])
     INSTALL_REQUIRES.extend(EXTRAS_REQUIRE["tf"])
     sys.argv.remove("--tf")