ruff check

prishajain1 · prishajain1 · commit f9e842a11fcd · 2026-01-21T00:40:31.000+05:30
diff --git a/src/maxdiffusion/pipelines/wan/wan_pipeline.py b/src/maxdiffusion/pipelines/wan/wan_pipeline.py
@@ -532,7 +532,7 @@ def prepare_latents_i2v_base(
       with self.mesh, nn_partitioning.axis_rules(self.config.logical_axis_rules):
           sharding_spec = P(self.config.mesh_axes[0], None, None, None, None)
           video_condition = jax.lax.with_sharding_constraint(
-            video_condition, 
+            video_condition,
             sharding_spec
           )
           encoded_output = self.vae.encode(video_condition, self.vae_cache)[0].mode()

Original file line number	Diff line number	Diff line change
`@@ -532,7 +532,7 @@ def prepare_latents_i2v_base(`
`532`	`532`	`with self.mesh, nn_partitioning.axis_rules(self.config.logical_axis_rules):`
`533`	`533`	`sharding_spec = P(self.config.mesh_axes[0], None, None, None, None)`
`534`	`534`	`video_condition = jax.lax.with_sharding_constraint(`
`535`		`- video_condition,`
	`535`	`+ video_condition,`
`536`	`536`	`sharding_spec`
`537`	`537`	`)`
`538`	`538`	`encoded_output = self.vae.encode(video_condition, self.vae_cache)[0].mode()`