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

Mark flaky tests (#25463)

Browse Source 
Make CI less brittle
This commit is contained in:
amyeroberts
2023-08-11 15:26:45 +01:00
committed by GitHub
parent 11757e2bbd
commit 5e5fa0d88c
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/wav2vec2/test_modeling_tf_wav2vec2.py
View File

@@ -306,7 +306,7 @@ class TFWav2Vec2ModelTester:
model = TFWav2Vec2ForCTC(config)
input_lengths = tf.constant([input_values.shape[-1] // i for i in [4, 2, 1]])
max_length_labels = model.wav2vec2._get_feat_extract_output_lengths(input_lengths)
labels = ids_tensor((input_values.shape[0], min(max_length_labels) - 1), model.config.vocab_size + 100)
labels = ids_tensor((input_values.shape[0], min(max_length_labels) - 1), model.config.vocab_size + 500)
with pytest.raises(ValueError):
model(input_values, labels=labels)