fix

huggingface · Dec 7, 2023 · f809550 · f809550
1 parent 4ea1f4d
commit f809550
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/pipelines/test_pipelines_fill_mask.py b/tests/pipelines/test_pipelines_fill_mask.py
@@ -232,8 +232,8 @@ def run_large_test(self, unmasker):
         self.assertEqual(
             [{k: x[k] for k in x if k != "sequence"} for x in simplified],
             [
-                {"score": 0.281867, "token": 6, "token_str": ","},
-                {"score": 0.095432, "token": 46686, "token_str": ":,"},
+                {"score": 0.2819, "token": 6, "token_str": ","},
+                {"score": 0.0954, "token": 46686, "token_str": ":,"},
             ],
         )