allenai · dirkgr · May 26, 2022 · May 25, 2022 · May 25, 2022 · May 26, 2022
diff --git a/allennlp/common/cached_transformers.py b/allennlp/common/cached_transformers.py
@@ -170,7 +170,7 @@ def strip_prefix(s):
             elif all(isinstance(x, str) for x in reinit_modules):
                 for regex in reinit_modules:
                     for name, module in transformer.named_modules():
-                        if re.search(regex, name):
+                        if re.search(str(regex), name):
                             module.apply(transformer._init_weights)
             else:
                 raise ValueError(

diff --git a/dev-requirements.txt b/dev-requirements.txt
@@ -4,7 +4,7 @@
 flake8
 
 # Static type checking
-mypy==0.950
+mypy==0.960
 
 # Automatic code formatting
 black==22.3.0

diff --git a/requirements.txt b/requirements.txt
@@ -31,6 +31,9 @@ base58
 # Spacy depends on typer, and typer had a bug. This is how we make sure we get the fixed version of typer.
 typer>=0.4.1
 
+# Protobuf is a dependency of wandb and tensorboard, but they are missing this pin.
+protobuf<=3.20.0
+
 ##################################################
 ###### Extra dependencies for integrations #######
 ##################################################