From 9e5b549b4d47678bdc74bc8f650e82cf25bfc245 Mon Sep 17 00:00:00 2001 From: sshleifer Date: Tue, 4 Feb 2020 14:09:45 -0500 Subject: [PATCH] fix default getattr --- tests/test_modeling_common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py index c49acf2316..a500351958 100644 --- a/tests/test_modeling_common.py +++ b/tests/test_modeling_common.py @@ -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)