SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix tests skip (#32012)

Browse Source 
* [run-slow] clip

* [run-slow] clip

* Fix skip -> skipTest

* [run-slow] clip
This commit is contained in:
Pavel Iakubovskii
2024-07-17 08:37:43 +01:00
committed by GitHub
parent 24cfcc2114
commit 691586b0dc
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/big_bird/test_modeling_big_bird.py
View File

@@ -716,7 +716,7 @@ class BigBirdModelIntegrationTest(unittest.TestCase):
""" """
if not self.test_attention_probs: if not self.test_attention_probs:
self.skip("test_attention_probs is set to False") self.skipTest("test_attention_probs is set to False")
model = BigBirdModel.from_pretrained( model = BigBirdModel.from_pretrained(
"google/bigbird-roberta-base", attention_type="block_sparse", num_random_blocks=3, block_size=16 "google/bigbird-roberta-base", attention_type="block_sparse", num_random_blocks=3, block_size=16

2
tests/models/whisper/test_modeling_whisper.py
View File

@@ -3335,7 +3335,7 @@ class WhisperEncoderModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.
fx_model_class_name = "Flax" + model_class.__name__ fx_model_class_name = "Flax" + model_class.__name__
if not hasattr(transformers, fx_model_class_name): if not hasattr(transformers, fx_model_class_name):
self.skip("Flax model does not exist") self.skipTest("Flax model does not exist")
# Output all for aggressive testing # Output all for aggressive testing
config.output_hidden_states = True config.output_hidden_states = True

2
tests/test_tokenization_common.py
View File

@@ -2678,7 +2678,7 @@ class TokenizerTesterMixin:
config = config_class() config = config_class()
if config.is_encoder_decoder or config.pad_token_id is None: if config.is_encoder_decoder or config.pad_token_id is None:
self.skip("Model is not an encoder-decoder model or has no set pad token id") self.skipTest("Model is not an encoder-decoder model or has no set pad token id")
# Build sequence # Build sequence
first_ten_tokens = list(tokenizer.get_vocab().keys())[:10] first_ten_tokens = list(tokenizer.get_vocab().keys())[:10]