From 56b33cc58c748041794d57bddf4f2748ebe52df9 Mon Sep 17 00:00:00 2001 From: Lysandre Date: Mon, 7 Jun 2021 17:08:58 +0200 Subject: [PATCH] Style --- src/transformers/dependency_versions_table.py | 2 +- tests/test_pipelines_text_classification.py | 7 ++++--- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index 27cb37f84f6ef2..98267b3a3e74c9 100644 --- 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 index 72c3fa30663a8f..698487a7aec75a 100644 --- 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))