fix default getattr

This commit is contained in:
sshleifer 2020-02-04 14:09:45 -05:00 committed by Lysandre Debut
parent 25848a6094
commit 9e5b549b4d

View File

@ -117,7 +117,7 @@ class ModelTesterMixin:
def test_attention_outputs(self):
config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()
seq_len = self.model_tester.seq_length
seq_len = getattr(self.model_tester, "seq_length", None)
decoder_seq_length = getattr(self.model_tester, "decoder_seq_length", seq_len)
encoder_seq_length = getattr(self.model_tester, "encoder_seq_length", seq_len)
decoder_key_length = getattr(self.model_tester, "key_length", decoder_seq_length)