Update BertForSyntaxParsing.py

BertForSyntaxParsing.py

@@ -167,6 +167,8 @@ def parse_logits(input_ids: List[List[int]], sentences: List[str], tokenizer: Be
 
     special_toks = tokenizer.all_special_tokens
     special_toks.remove(tokenizer.unk_token)
+    special_toks.remove(tokenizer.mask_token)
+    
     for i in range(len(sentences)):
         deps = logits.dependency_head_indices[i].tolist()
         funcs = logits.function_logits.argmax(-1)[i].tolist()