double quotes

This commit is contained in:
sshleifer
2020-02-04 13:58:05 -05:00
committed by Lysandre Debut
parent cbcb83f21d
commit 25848a6094

View File

@@ -118,8 +118,8 @@ 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
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_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)
encoder_key_length = getattr(self.model_tester, "key_length", encoder_seq_length)