Pin protobouf that breaks TensorBoard in PyTorch (#17440)

2022-05-26 09:56:55 -04:00
parent 2295bcaea8
commit 7535d92e71
2 changed files with 3 additions and 2 deletions
--- a/setup.py
+++ b/setup.py
@@ -132,7 +132,7 @@ _deps = [
    "packaging>=20.0",
    "parameterized",
    "phonemizer",
-    "protobuf",
+    "protobuf<=3.20.1",
    "psutil",
    "pyyaml>=5.1",
    "pydantic",
@@ -293,6 +293,7 @@ extras["testing"] = (
        "nltk",
        "GitPython",
        "hf-doc-builder",
+        "protobuf", # Can be removed once we can unpin protobuf
        "sacremoses",
        "rjieba"
    )