"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "536ea2aca234fb48c5c69769431d643b0d93b233"
Unverified Commit 2adc8c92 authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

W2v2 test require torch (#10665)

* Adds a @require_torch to a test that requires it

* Tokenizer too

* Style
parent 055ed78f
...@@ -21,7 +21,7 @@ import unittest ...@@ -21,7 +21,7 @@ import unittest
import numpy as np import numpy as np
from transformers import WAV_2_VEC_2_PRETRAINED_MODEL_ARCHIVE_LIST, Wav2Vec2Config, Wav2Vec2FeatureExtractor from transformers import WAV_2_VEC_2_PRETRAINED_MODEL_ARCHIVE_LIST, Wav2Vec2Config, Wav2Vec2FeatureExtractor
from transformers.testing_utils import slow from transformers.testing_utils import require_torch, slow
from .test_sequence_feature_extraction_common import SequenceFeatureExtractionTestMixin from .test_sequence_feature_extraction_common import SequenceFeatureExtractionTestMixin
...@@ -134,6 +134,7 @@ class Wav2Vec2FeatureExtractionTest(SequenceFeatureExtractionTestMixin, unittest ...@@ -134,6 +134,7 @@ class Wav2Vec2FeatureExtractionTest(SequenceFeatureExtractionTestMixin, unittest
_check_zero_mean_unit_variance(input_values[2]) _check_zero_mean_unit_variance(input_values[2])
@slow @slow
@require_torch
def test_pretrained_checkpoints_are_set_correctly(self): def test_pretrained_checkpoints_are_set_correctly(self):
# this test makes sure that models that are using # this test makes sure that models that are using
# group norm don't have their feature extractor return the # group norm don't have their feature extractor return the
......
...@@ -30,7 +30,7 @@ from transformers import ( ...@@ -30,7 +30,7 @@ from transformers import (
Wav2Vec2Tokenizer, Wav2Vec2Tokenizer,
) )
from transformers.models.wav2vec2.tokenization_wav2vec2 import VOCAB_FILES_NAMES from transformers.models.wav2vec2.tokenization_wav2vec2 import VOCAB_FILES_NAMES
from transformers.testing_utils import slow from transformers.testing_utils import require_torch, slow
from .test_tokenization_common import TokenizerTesterMixin from .test_tokenization_common import TokenizerTesterMixin
...@@ -340,6 +340,7 @@ class Wav2Vec2TokenizerTest(unittest.TestCase): ...@@ -340,6 +340,7 @@ class Wav2Vec2TokenizerTest(unittest.TestCase):
self.assertListEqual(processed.attention_mask.sum(-1).tolist(), [800, 1000, 1200]) self.assertListEqual(processed.attention_mask.sum(-1).tolist(), [800, 1000, 1200])
@slow @slow
@require_torch
def test_pretrained_checkpoints_are_set_correctly(self): def test_pretrained_checkpoints_are_set_correctly(self):
# this test makes sure that models that are using # this test makes sure that models that are using
# group norm don't have their tokenizer return the # group norm don't have their tokenizer return the
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment