ruff formatting

yuhezhang-ai · yuhezhang-ai · commit e034fd99cd58 · 2026-01-05T11:24:39.000-05:00
Signed-off-by: Yuhe Zhang &lt;yuhe@polarr.co&gt;
diff --git a/nemo_automodel/components/_peft/lora_moe.py b/nemo_automodel/components/_peft/lora_moe.py
@@ -492,9 +492,7 @@ def to_local(proj):
                 trans_b=False,
             )
             # [T, R] @ [E_local, R, H] -> [T, H]
-            lora_out1 = ops.gmm(
-                lora_out1_A, to_local(self.lora_gate_and_up_B), tokens_per_expert, trans_b=False
-            )
+            lora_out1 = ops.gmm(lora_out1_A, to_local(self.lora_gate_and_up_B), tokens_per_expert, trans_b=False)
 
             output1 = output1 + lora_out1 * self.scale
 

Original file line number	Diff line number	Diff line change
`@@ -492,9 +492,7 @@ def to_local(proj):`
`492`	`492`	`trans_b=False,`
`493`	`493`	`)`
`494`	`494`	`# [T, R] @ [E_local, R, H] -> [T, H]`
`495`		`- lora_out1 = ops.gmm(`
`496`		`- lora_out1_A, to_local(self.lora_gate_and_up_B), tokens_per_expert, trans_b=False`
`497`		`- )`
	`495`	`+ lora_out1 = ops.gmm(lora_out1_A, to_local(self.lora_gate_and_up_B), tokens_per_expert, trans_b=False)`
`498`	`496`
`499`	`497`	`output1 = output1 + lora_out1 * self.scale`
`500`	`498`