Restore pipeline test to main

mydatascience · mydatascience · commit fca895835549 · 2025-11-12T22:27:27.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
@@ -231,7 +231,6 @@ 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
`@@ -231,7 +231,6 @@ def test_circular_deepseek_megablox_same_output_and_grad(self):`
`231`	`231`	`sparse_matmul=False,`
`232`	`232`	`capacity_factor=1,`
`233`	`233`	`decoder_block="deepseek",`
`234`		`- attention_type="mla",`
`235`	`234`	`)`
`236`	`235`	`self.assert_pipeline_same_output_and_grad(config, single_pipeline_stage_class=deepseek.DeepSeekMoELayer)`
`237`	`236`