ignore transformers warnings

huggingface · May 23, 2024 · 889a48d · 889a48d
1 parent 86953e5
commit 889a48d
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 4 deletions.
diff --git a/tests/test_fingerprint.py b/tests/test_fingerprint.py
@@ -2,6 +2,7 @@
 import os
 import pickle
 import subprocess
+import warnings
 from functools import partial
 from pathlib import Path
 from tempfile import gettempdir
@@ -87,15 +88,21 @@ def encode(x):
             return tokenizer(x)
 
         # TODO: add hash consistency tests across sessions
-        tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
+        with warnings.catch_warnings():
+            warnings.simplefilter("ignore")
+            tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
         hash1 = Hasher.hash(tokenizer)
         hash1_lambda = Hasher.hash(lambda x: tokenizer(x))
         hash1_encode = Hasher.hash(encode)
-        tokenizer = AutoTokenizer.from_pretrained("bert-base-cased")
+        with warnings.catch_warnings():
+            warnings.simplefilter("ignore")
+            tokenizer = AutoTokenizer.from_pretrained("bert-base-cased")
         hash2 = Hasher.hash(tokenizer)
         hash2_lambda = Hasher.hash(lambda x: tokenizer(x))
         hash2_encode = Hasher.hash(encode)
-        tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
+        with warnings.catch_warnings():
+            warnings.simplefilter("ignore")
+            tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
         hash3 = Hasher.hash(tokenizer)
         hash3_lambda = Hasher.hash(lambda x: tokenizer(x))
         hash3_encode = Hasher.hash(encode)

diff --git a/tests/test_metric_common.py b/tests/test_metric_common.py
@@ -18,6 +18,7 @@
 import inspect
 import os
 import re
+import warnings
 from contextlib import contextmanager
 from functools import wraps
 from unittest.mock import patch
@@ -105,7 +106,8 @@ def test_load_metric(self, metric_name):
         parameters = inspect.signature(metric._compute).parameters
         self.assertTrue(all(p.kind != p.VAR_KEYWORD for p in parameters.values()))  # no **kwargs
         # run doctest
-        with self.patch_intensive_calls(metric_name, metric_module.__name__):
+        with self.patch_intensive_calls(metric_name, metric_module.__name__), warnings.catch_warnings():
+            warnings.simplefilter("ignore")
             with self.use_local_metrics():
                 try:
                     results = doctest.testmod(metric_module, verbose=True, raise_on_error=True)