Suppress tokenizer warning

Author: Aethor

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "tibert"
-version = "0.0.2"
+version = "0.0.3"
 description = "BERT for Coreference Resolution"
 authors = ["Arthur Amalvy <[email protected]>"]
 license = "GPL-3.0-only"

tibert/bertcoref.py

@@ -286,6 +286,10 @@ def torch_call(self, features) -> Union[dict, BatchEncoding]:
             coref_labels and mention_labels
         )
 
+        warning_state = self.tokenizer.deprecation_warnings.get(
+            "Asking-to-pad-a-fast-tokenizer", False
+        )
+        self.tokenizer.deprecation_warnings["Asking-to-pad-a-fast-tokenizer"] = True
         batch = self.tokenizer.pad(
             features,
             padding=self.padding,
@@ -294,6 +298,9 @@ def torch_call(self, features) -> Union[dict, BatchEncoding]:
             # they are not of the same length yet.
             return_tensors="pt" if coref_labels is None else None,
         )
+        self.tokenizer.deprecation_warnings[
+            "Asking-to-pad-a-fast-tokenizer"
+        ] = warning_state
 
         # keep encoding info
         batch._encodings = [f.encodings[0] for f in features]