Skip to content

GitLab

Unverified Commit 1a647abf authored by Sam Shleifer's avatar Sam Shleifer Committed by GitHub
Browse files

[fix] check code quality (#5772)

parent b23d3a5a
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 7 deletions
examples/seq2seq/utils.py
View file @ 1a647abf
...@@ -182,7 +182,6 @@ class SortishSampler(Sampler): ...@@ -182,7 +182,6 @@ class SortishSampler(Sampler):
return iter(sort_idx) return iter(sort_idx)
logger = getLogger(__name__) logger = getLogger(__name__)
......
tests/test_modeling_t5.py
View file @ 1a647abf
...@@ -143,12 +143,6 @@ class T5ModelTester: ...@@ -143,12 +143,6 @@ class T5ModelTester:
decoder_attention_mask=decoder_attention_mask, decoder_attention_mask=decoder_attention_mask,
) )
decoder_output, decoder_past, encoder_output = model(input_ids=input_ids, decoder_input_ids=decoder_input_ids) 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(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(decoder_output.size(), (self.batch_size, self.decoder_seq_length, self.hidden_size))
self.parent.assertEqual(len(decoder_past), 2) self.parent.assertEqual(len(decoder_past), 2)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment