From 91003c9ead70eef76147b2c49638f410ab74307c Mon Sep 17 00:00:00 2001 From: Akihiro Nitta Date: Sun, 29 May 2022 06:30:42 +0900 Subject: [PATCH] Pin protobuf version (#13177) --- pytorch_lightning/setup_tools.py | 3 ++- requirements/base.txt | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/pytorch_lightning/setup_tools.py b/pytorch_lightning/setup_tools.py index 95ca568bf8..2df3c7946b 100644 --- a/pytorch_lightning/setup_tools.py +++ b/pytorch_lightning/setup_tools.py @@ -32,9 +32,10 @@ def _load_requirements( reqs = [] for ln in lines: # filer all comments + comment = "" if comment_char in ln: + comment = ln[ln.index(comment_char) :] ln = ln[: ln.index(comment_char)] - comment = ln[ln.index(comment_char) :] if comment_char in ln else "" req = ln.strip() # skip directly installed dependencies if not req or req.startswith("http") or "@http" in req: diff --git a/requirements/base.txt b/requirements/base.txt index c26c199115..555997c657 100644 --- a/requirements/base.txt +++ b/requirements/base.txt @@ -8,3 +8,4 @@ torchmetrics>=0.4.1, <=0.7.2 pyDeprecate>=0.3.1, <=0.3.2 packaging>=17.0, <=21.3 typing-extensions>=4.0.0, <4.2.1 +protobuf<=3.20.1 # strict. TODO: Remove after tensorboard gets compatible https://github.com/tensorflow/tensorboard/issues/5708