From 4fe9716a793cf077e07f21d0da0b9332d986dd2d Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Mon, 12 Jun 2023 18:14:15 +0200 Subject: [PATCH] Skip RWKV test in past CI (#24204) * fix * fix --------- Co-authored-by: ydshieh --- src/transformers/pytorch_utils.py | 1 + tests/models/rwkv/test_modeling_rwkv.py | 9 +++++++++ 2 files changed, 10 insertions(+) diff --git a/src/transformers/pytorch_utils.py b/src/transformers/pytorch_utils.py index 6c5fc5f4a4..f076eb445c 100644 --- a/src/transformers/pytorch_utils.py +++ b/src/transformers/pytorch_utils.py @@ -28,6 +28,7 @@ logger = logging.get_logger(__name__) parsed_torch_version_base = version.parse(version.parse(torch.__version__).base_version) +is_torch_greater_or_equal_than_2_0 = parsed_torch_version_base >= version.parse("2.0") is_torch_greater_or_equal_than_1_10 = parsed_torch_version_base >= version.parse("1.10") is_torch_less_than_1_11 = parsed_torch_version_base < version.parse("1.11") diff --git a/tests/models/rwkv/test_modeling_rwkv.py b/tests/models/rwkv/test_modeling_rwkv.py index 638180cbcf..67430f6f3d 100644 --- a/tests/models/rwkv/test_modeling_rwkv.py +++ b/tests/models/rwkv/test_modeling_rwkv.py @@ -34,6 +34,9 @@ if is_torch_available(): RwkvForCausalLM, RwkvModel, ) + from transformers.pytorch_utils import is_torch_greater_or_equal_than_2_0 +else: + is_torch_greater_or_equal_than_2_0 = False class RwkvModelTester: @@ -258,6 +261,9 @@ class RwkvModelTester: return config, inputs_dict +@unittest.skipIf( + not is_torch_greater_or_equal_than_2_0, reason="See https://github.com/huggingface/transformers/pull/24204" +) @require_torch class RwkvModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixin, unittest.TestCase): all_model_classes = (RwkvModel, RwkvForCausalLM) if is_torch_available() else () @@ -418,6 +424,9 @@ class RwkvModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixin self.assertIsNotNone(model) +@unittest.skipIf( + not is_torch_greater_or_equal_than_2_0, reason="See https://github.com/huggingface/transformers/pull/24204" +) @slow class RWKVIntegrationTests(unittest.TestCase): def setUp(self):