From 25848a6094179391b03b34d447baa3aad4d815af Mon Sep 17 00:00:00 2001 From: sshleifer Date: Tue, 4 Feb 2020 13:58:05 -0500 Subject: [PATCH] double quotes --- tests/test_modeling_common.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py index 06dcf3199c..c49acf2316 100644 --- a/tests/test_modeling_common.py +++ b/tests/test_modeling_common.py @@ -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)