From 1a647abf0b25f7f3f110272a7da92257f030620c Mon Sep 17 00:00:00 2001 From: Sam Shleifer Date: Wed, 15 Jul 2020 14:59:38 -0400 Subject: [PATCH] [fix] check code quality (#5772) --- examples/seq2seq/utils.py | 1 - tests/test_modeling_t5.py | 6 ------ 2 files changed, 7 deletions(-) diff --git a/examples/seq2seq/utils.py b/examples/seq2seq/utils.py index 8b4388f442..0938f0f373 100644 --- a/examples/seq2seq/utils.py +++ b/examples/seq2seq/utils.py @@ -182,7 +182,6 @@ class SortishSampler(Sampler): return iter(sort_idx) - logger = getLogger(__name__) diff --git a/tests/test_modeling_t5.py b/tests/test_modeling_t5.py index 6c4054dc46..9c5cd9dbce 100644 --- a/tests/test_modeling_t5.py +++ b/tests/test_modeling_t5.py @@ -143,12 +143,6 @@ class T5ModelTester: decoder_attention_mask=decoder_attention_mask, ) decoder_output, decoder_past, encoder_output = model(input_ids=input_ids, decoder_input_ids=decoder_input_ids) - - result = { - "encoder_output": encoder_output, - "decoder_output": decoder_output, - "decoder_past": decoder_past, - } self.parent.assertEqual(encoder_output.size(), (self.batch_size, self.encoder_seq_length, self.hidden_size)) self.parent.assertEqual(decoder_output.size(), (self.batch_size, self.decoder_seq_length, self.hidden_size)) self.parent.assertEqual(len(decoder_past), 2)