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

[Tests] Improve flax test_attention_outputs (#21486)

Browse Source 
improving flax tests
This commit is contained in:
Shubhamai
2023-02-10 22:01:49 +05:30
committed by GitHub
parent c88b11c591
commit 9e40bba6ba
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_modeling_flax_common.py
View File

@@ -650,6 +650,9 @@ class FlaxModelTesterMixin:
check_hidden_states_output(inputs_dict, config, model_class) check_hidden_states_output(inputs_dict, config, model_class)
def test_attention_outputs(self): def test_attention_outputs(self):
if not self.has_attentions:
self.skipTest(reason="Model does not output attentions")
config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common() config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()
config.return_dict = True config.return_dict = True