make fixup

gante · gante · commit f59bbb7cbb59 · 2022-07-22T11:27:52.000Z
diff --git a/tests/models/encoder_decoder/test_modeling_encoder_decoder.py b/tests/models/encoder_decoder/test_modeling_encoder_decoder.py
@@ -443,7 +443,10 @@ def check_encoder_decoder_model_output_attentions_from_config(
             decoder_attention_mask=decoder_attention_mask,
         )
         self.assertTrue(
-            all(key not in outputs_encoder_decoder for key in ["encoder_attentions", "decoder_attentions", "cross_attentions"])
+            all(
+                key not in outputs_encoder_decoder
+                for key in ["encoder_attentions", "decoder_attentions", "cross_attentions"]
+            )
         )
 
         config.output_attentions = True  # inner model config -> will work
diff --git a/tests/models/encoder_decoder/test_modeling_tf_encoder_decoder.py b/tests/models/encoder_decoder/test_modeling_tf_encoder_decoder.py
@@ -345,7 +345,10 @@ def check_encoder_decoder_model_output_attentions_from_config(
             kwargs=kwargs,
         )
         self.assertTrue(
-            all(key not in outputs_encoder_decoder for key in ["encoder_attentions", "decoder_attentions", "cross_attentions"])
+            all(
+                key not in outputs_encoder_decoder
+                for key in ["encoder_attentions", "decoder_attentions", "cross_attentions"]
+            )
         )
 
         config.output_attentions = True  # inner model config -> will work

Original file line number	Diff line number	Diff line change
`@@ -443,7 +443,10 @@ def check_encoder_decoder_model_output_attentions_from_config(`
`443`	`443`	`decoder_attention_mask=decoder_attention_mask,`
`444`	`444`	`)`
`445`	`445`	`self.assertTrue(`
`446`		`- all(key not in outputs_encoder_decoder for key in ["encoder_attentions", "decoder_attentions", "cross_attentions"])`
	`446`	`+ all(`
	`447`	`+ key not in outputs_encoder_decoder`
	`448`	`+ for key in ["encoder_attentions", "decoder_attentions", "cross_attentions"]`
	`449`	`+ )`
`447`	`450`	`)`
`448`	`451`
`449`	`452`	`config.output_attentions = True # inner model config -> will work`
Original file line number	Diff line number	Diff line change
`@@ -345,7 +345,10 @@ def check_encoder_decoder_model_output_attentions_from_config(`
`345`	`345`	`kwargs=kwargs,`
`346`	`346`	`)`
`347`	`347`	`self.assertTrue(`
`348`		`- all(key not in outputs_encoder_decoder for key in ["encoder_attentions", "decoder_attentions", "cross_attentions"])`
	`348`	`+ all(`
	`349`	`+ key not in outputs_encoder_decoder`
	`350`	`+ for key in ["encoder_attentions", "decoder_attentions", "cross_attentions"]`
	`351`	`+ )`
`349`	`352`	`)`
`350`	`353`
`351`	`354`	`config.output_attentions = True # inner model config -> will work`