Fix linter

mydatascience · mydatascience · commit 3b4b943d60b5 · 2025-11-08T00:11:36.000+04:00
Signed-off-by: Vladimir Suvorov &lt;suvorovv@google.com&gt;
diff --git a/tests/pipeline_parallelism_test.py b/tests/pipeline_parallelism_test.py
@@ -239,6 +239,7 @@ def test_circular_deepseek_megablox_same_output_and_grad(self):
         sparse_matmul=False,
         capacity_factor=1,
         decoder_block="deepseek",
+        attention_type="mla",
     )
     self.assert_pipeline_same_output_and_grad(config, single_pipeline_stage_class=deepseek.DeepSeekMoELayer)
 

Original file line number	Diff line number	Diff line change
`@@ -239,6 +239,7 @@ def test_circular_deepseek_megablox_same_output_and_grad(self):`
`239`	`239`	`sparse_matmul=False,`
`240`	`240`	`capacity_factor=1,`
`241`	`241`	`decoder_block="deepseek",`
	`242`	`+ attention_type="mla",`
`242`	`243`	`)`
`243`	`244`	`self.assert_pipeline_same_output_and_grad(config, single_pipeline_stage_class=deepseek.DeepSeekMoELayer)`
`244`	`245`