Skip to content

GitLab

Unverified Commit 9e40bba6 authored by Shubhamai's avatar Shubhamai Committed by GitHub
Browse files

[Tests] Improve flax test_attention_outputs (#21486) 

improving flax tests
parent c88b11c5
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
tests/test_modeling_flax_common.py
View file @ 9e40bba6
...@@ -650,6 +650,9 @@ class FlaxModelTesterMixin: ...@@ -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
......
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