Style

huggingface · Jun 7, 2021 · 56b33cc · 56b33cc
1 parent e5347a9
commit 56b33cc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py
@@ -21,7 +21,7 @@
     "ipadic": "ipadic>=1.0.0,<2.0",
     "isort": "isort>=5.5.4",
     "jax": "jax>=0.2.8",
-    "jaxlib": "jaxlib>=0.1.59",
+    "jaxlib": "jaxlib>=0.1.65",
     "jieba": "jieba",
     "keras2onnx": "keras2onnx",
     "nltk": "nltk",

diff --git a/tests/test_pipelines_text_classification.py b/tests/test_pipelines_text_classification.py
@@ -13,13 +13,14 @@
 # limitations under the License.
 
 import unittest
-import numpy as np
 
-from .test_pipelines_common import MonoInputPipelineCommonMixin
+import numpy as np
 
 from transformers import AutoTokenizer, DistilBertConfig, DistilBertForSequenceClassification, pipeline
 from transformers.testing_utils import slow
 
+from .test_pipelines_common import MonoInputPipelineCommonMixin
+
 
 class TextClassificationPipelineTests(MonoInputPipelineCommonMixin, unittest.TestCase):
     pipeline_task = "sentiment-analysis"
@@ -120,7 +121,7 @@ def pipeline_model_argument(argument=None):
     @slow
     def test_function_to_apply_error(self):
         for model_name in self.small_models:
-            string_input, _ = VALID_INPUTS
+            string_input = "I really disagree with what you've said."
             tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=True)
 
             model = DistilBertForSequenceClassification(DistilBertConfig.from_pretrained(model_name))