From 83dc5762e756d399dfcab314128efe213c264e83 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Tue, 20 Jun 2023 18:35:45 +0200 Subject: [PATCH] Skip a tapas (tokenization) test in past CI (#24378) fix Co-authored-by: ydshieh --- tests/models/tapas/test_tokenization_tapas.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/tests/models/tapas/test_tokenization_tapas.py b/tests/models/tapas/test_tokenization_tapas.py index d3af3276f7..b4cca18162 100644 --- a/tests/models/tapas/test_tokenization_tapas.py +++ b/tests/models/tapas/test_tokenization_tapas.py @@ -22,7 +22,7 @@ from typing import List import numpy as np import pandas as pd -from transformers import AddedToken +from transformers import AddedToken, is_torch_available from transformers.models.tapas.tokenization_tapas import ( VOCAB_FILES_NAMES, BasicTokenizer, @@ -44,6 +44,12 @@ from transformers.testing_utils import ( from ...test_tokenization_common import TokenizerTesterMixin, filter_non_english, merge_model_tokenizer_mappings +if is_torch_available(): + from transformers.pytorch_utils import is_torch_greater_or_equal_than_1_12 +else: + is_torch_greater_or_equal_than_1_12 = False + + @require_tokenizers @require_pandas class TapasTokenizationTest(TokenizerTesterMixin, unittest.TestCase): @@ -1026,6 +1032,7 @@ class TapasTokenizationTest(TokenizerTesterMixin, unittest.TestCase): # Do the same test as modeling common. self.assertIn(0, output["token_type_ids"][0]) + @unittest.skipIf(not is_torch_greater_or_equal_than_1_12, reason="Tapas is only available in torch v1.12+") @require_torch @slow def test_torch_encode_plus_sent_to_model(self):