Run pylinter

Author: Rohan-Bierneni

dags/sparsity_diffusion_devx/maxtext_moe_tpu_e2e.py

@@ -76,7 +76,7 @@
       "qwen3-next-80b": {
           "script_name": "tpu/qwen/next/qwen3-next-80b-a3b/2_test_qwen3_next_80b_a3b",
           "cluster": XpkClusters.TPU_V5P_128_CLUSTER,
-          "time_out_in_min": 90, 
+          "time_out_in_min": 90,
       },
   }
 

dags/sparsity_diffusion_devx/qwen3_next_test.py

@@ -36,7 +36,6 @@
     start_date=datetime.datetime(2024, 1, 1),
     catchup=False,
 ) as dag:
-  
   # Your specified custom docker image
   custom_docker_image = "gcr.io/tpu-prod-env-multipod/maxtext_stable_stack_candidate:rbierneni-qwen-test"
 
@@ -48,9 +47,9 @@
           f"export HF_TOKEN={HF_TOKEN}; export BASE_OUTPUT_PATH=$GCS_OUTPUT; bash tests/end_to_end/tpu/qwen/next/qwen3-next-80b-a3b/2_test_qwen3_next_80b_a3b.sh",
       ),
       docker_image=custom_docker_image,
-      test_owner=test_owner.ROHAN_B, # Update the owner if necessary
+      test_owner=test_owner.ROHAN_B,  # Update the owner if necessary
       cluster=XpkClusters.TPU_V5P_128_CLUSTER,
   ).run()
 
   # No chained dependencies required since there is only one task
-  test_qwen3_next
+  test_qwen3_next