Fix quotes in ignore_keys (#473)

castorini · Apr 11, 2021 · 7e9bf7e · 7e9bf7e
1 parent 587fd7b
commit 7e9bf7e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyserini/dsearch/_model.py b/pyserini/dsearch/_model.py
@@ -12,7 +12,7 @@ class AnceEncoder(PreTrainedModel):
     base_model_prefix = 'ance_encoder'
     load_tf_weights = None
     _keys_to_ignore_on_load_missing = [r'position_ids']
-    _keys_to_ignore_on_load_unexpected = [r'pooler", r"classifier']
+    _keys_to_ignore_on_load_unexpected = [r'pooler', r'classifier']
 
     def __init__(self, config: RobertaConfig):
         requires_pytorch(self)