Skip to content

Commit 7f73371

Browse files
Iman GohariLuca-Calabria
authored andcommitted
Added skip for unsuported tests for mistral/mixtral (huggingface#1462)
1 parent 19aeeba commit 7f73371

2 files changed

Lines changed: 80 additions & 0 deletions

File tree

tests/transformers/tests/models/mistral/test_modeling_mistral.py

Lines changed: 40 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -297,6 +297,46 @@ class MistralModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCas
297297
test_headmasking = False
298298
test_pruning = False
299299

300+
@unittest.skip(reason="This test is not supported for Mistral")
301+
def test_beam_search_generate(self):
302+
pass
303+
304+
@unittest.skip(reason="This test is not supported for Mistral")
305+
def test_beam_search_generate_dict_output(self):
306+
pass
307+
308+
@unittest.skip(reason="This test is not supported for Mistral")
309+
def test_beam_search_generate_dict_outputs_use_cache(self):
310+
pass
311+
312+
@unittest.skip(reason="This test is not supported for Mistral")
313+
def test_attention_outputs(self):
314+
pass
315+
316+
@unittest.skip(reason="This test is not supported for Mistral")
317+
def test_constrained_beam_search_generate(self):
318+
pass
319+
320+
@unittest.skip(reason="This test is not supported for Mistral")
321+
def test_constrained_beam_search_generate_dict_output(self):
322+
pass
323+
324+
@unittest.skip(reason="This test is not supported for Mistral")
325+
def test_contrastive_generate_dict_outputs_use_cache(self):
326+
pass
327+
328+
@unittest.skip(reason="This test is not supported for Mistral")
329+
def test_greedy_generate_dict_outputs(self):
330+
pass
331+
332+
@unittest.skip(reason="This test is not supported for Mistral")
333+
def test_greedy_generate_dict_outputs_use_cache(self):
334+
pass
335+
336+
@unittest.skip(reason="This test is not supported for Mistral")
337+
def test_sample_generate_dict_output(self):
338+
pass
339+
300340
# TODO (ydshieh): Check this. See https://app.circleci.com/pipelines/github/huggingface/transformers/79245/workflows/9490ef58-79c2-410d-8f51-e3495156cf9c/jobs/1012146
301341
def is_pipeline_test_to_skip(
302342
self, pipeline_test_casse_name, config_class, model_architecture, tokenizer_name, processor_name

tests/transformers/tests/models/mixtral/test_modeling_mixtral.py

Lines changed: 40 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -298,6 +298,46 @@ class MixtralModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCas
298298
test_headmasking = False
299299
test_pruning = False
300300

301+
@unittest.skip(reason="This test is not supported for Mixtral")
302+
def test_beam_search_generate(self):
303+
pass
304+
305+
@unittest.skip(reason="This test is not supported for Mixtral")
306+
def test_beam_search_generate_dict_output(self):
307+
pass
308+
309+
@unittest.skip(reason="This test is not supported for Mixtral")
310+
def test_beam_search_generate_dict_outputs_use_cache(self):
311+
pass
312+
313+
@unittest.skip(reason="This test is not supported for Mixtral")
314+
def test_attention_outputs(self):
315+
pass
316+
317+
@unittest.skip(reason="This test is not supported for Mixtral")
318+
def test_constrained_beam_search_generate_dict_output(self):
319+
pass
320+
321+
@unittest.skip(reason="This test is not supported for Mixtral")
322+
def test_contrastive_generate_dict_outputs_use_cache(self):
323+
pass
324+
325+
@unittest.skip(reason="This test is not supported for Mixtral")
326+
def test_greedy_generate_dict_outputs(self):
327+
pass
328+
329+
@unittest.skip(reason="This test is not supported for Mixtral")
330+
def test_greedy_generate_dict_outputs_use_cache(self):
331+
pass
332+
333+
@unittest.skip(reason="This test is not supported for Mixtral")
334+
def test_retain_grad_hidden_states_attentions(self):
335+
pass
336+
337+
@unittest.skip(reason="This test is not supported for Mixtral")
338+
def test_sample_generate_dict_output(self):
339+
pass
340+
301341
# TODO (ydshieh): Check this. See https://app.circleci.com/pipelines/github/huggingface/transformers/79245/workflows/9490ef58-79c2-410d-8f51-e3495156cf9c/jobs/1012146
302342
def is_pipeline_test_to_skip(
303343
self, pipeline_test_casse_name, config_class, model_architecture, tokenizer_name, processor_name

0 commit comments

Comments
 (0)