fix lint issues

csgoogle · csgoogle · commit 039ceb3e68a9 · 2026-04-09T11:59:20.000Z
diff --git a/src/maxdiffusion/models/wan/transformers/transformer_wan.py b/src/maxdiffusion/models/wan/transformers/transformer_wan.py
@@ -249,7 +249,7 @@ def __init__(
         kernel_init=nnx.with_partitioning(
             nnx.initializers.xavier_uniform(),
             (
-                 "mlp",
+                "mlp",
                 "embed",
             ),
         ),

Original file line number	Diff line number	Diff line change
`@@ -249,7 +249,7 @@ def __init__(`
`249`	`249`	`kernel_init=nnx.with_partitioning(`
`250`	`250`	`nnx.initializers.xavier_uniform(),`
`251`	`251`	`(`
`252`		`- "mlp",`
	`252`	`+ "mlp",`
`253`	`253`	`"embed",`
`254`	`254`	`),`
`255`	`255`	`),`