Skip to content

Commit 1f4f314

Browse files
committed
fix an issue where context masks default would be incorrect
1 parent 8218862 commit 1f4f314

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

renard/ner_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ def __init__(
9292
assert all(
9393
[len(cm) == len(elt) for elt, cm in zip(self.elements, context_mask)]
9494
)
95-
self._context_mask = context_mask or [[1] * len(elt) for elt in self.elements]
95+
self._context_mask = context_mask or [[0] * len(elt) for elt in self.elements]
9696

9797
self.tokenizer = tokenizer
9898

0 commit comments

Comments
 (0)