Merge pull request #305 from mozilla/fix_protobuf_version

Pin protobuf version to <= 3.20.1 Tensorboard is not compatible with newer versions of protobuf. Until tensorflow/tensorboard#5708 is resolved we need to pin protobuf to <= 3.20.1
mozilla · Jun 13, 2022 · 2b2c84e · 2b2c84e
2 parents 0d69bb7 + 523c644
commit 2b2c84e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cli/setup.py b/cli/setup.py
@@ -20,7 +20,8 @@
         'scikit-learn>=0.22.2',
         'selenium>=3.141.0',
         'tensorboardX>=1.6,<2.0',
-        'torch>=1.0,<2.0'
+        'torch>=1.0,<2.0',
+        'protobuf <= 3.20.1',
     ],
     dependency_links=[
         'https://download.pytorch.org/whl/cu110/torch_stable.html'