fix linting

Author: HadarIngonyama

src/maxdiffusion/generate_wan.py

@@ -392,4 +392,4 @@ def main(argv: Sequence[str]) -> None:
 
 if __name__ == "__main__":
   with transformer_engine_context():
-    app.run(main)
+    app.run(main)

src/maxdiffusion/pipelines/wan/wan_pipeline_2_2.py

@@ -840,4 +840,4 @@ def scan_body(carry, t):
       if profiler:
         latents.block_until_ready()
         profiler.stop()
-  return latents
+  return latents

src/maxdiffusion/tests/wan/wan2_2_magcache_test.py

@@ -386,4 +386,4 @@ def tearDownClass(cls):
 
 
 if __name__ == "__main__":
-  absltest.main()
+  absltest.main()