ruff fix

pablo-garay · pablo-garay · commit 0e3378f2fbaa · 2025-11-04T14:48:19.000-08:00
Signed-off-by: Pablo Garay &lt;pagaray@nvidia.com&gt;
diff --git a/.github/workflows/cicd-approve-test-queue.yml b/.github/workflows/cicd-approve-test-queue.yml
@@ -150,4 +150,3 @@ jobs:
             -H 'Content-type: application/json' \
             --data "{\"text\":\":robot_joy: <https://github.com/${GITHUB_REPOSITORY}/actions/runs/${GITHUB_RUN_ID}|Test-queue-approval-bot workflow> failed. Please review manually.\n\ncc ${SLACK_WEBHOOK_ADMIN}\"}" \
             $SLACK_WEBHOOK
-
diff --git a/README.md b/README.md
@@ -27,4 +27,4 @@ If you find our code useful, please consider citing the following papers:
   journal={arXiv preprint arXiv:2501.03575},
   year={2025}
 }
-```
+```
diff --git a/dfm/examples/Automodel/finetune/wan2_1_t2v_flow.yaml b/dfm/examples/Automodel/finetune/wan2_1_t2v_flow.yaml
@@ -56,4 +56,4 @@ checkpoint:
   checkpoint_dir: /opt/DFM/wan_t2v_flow_outputs_base_recipe_fsdp_run_1/
   model_save_format: torch_save
   save_consolidated: false
-  restore_from: null
+  restore_from: null
diff --git a/dfm/examples/Automodel/finetune/wan2_1_t2v_flow_multinode.yaml b/dfm/examples/Automodel/finetune/wan2_1_t2v_flow_multinode.yaml
@@ -56,4 +56,4 @@ checkpoint:
   checkpoint_dir: /opt/DFM/wan_t2v_flow_outputs_base_recipe_multi_node_fsdp_run_3/
   model_save_format: torch_save
   save_consolidated: false
-  restore_from: null
+  restore_from: null
diff --git a/dfm/src/Automodel/README.md b/dfm/src/Automodel/README.md
@@ -1,4 +1,3 @@
 # Automodel
 
 Models implemented using Dtensors (NeMo Automodel)
-
diff --git a/dfm/src/megatron/README.md b/dfm/src/megatron/README.md
@@ -1,4 +1,3 @@
 # MBridge
 
 Megatron models including DiT and dLLM implementations.
-
diff --git a/dfm/src/megatron/base/README.md b/dfm/src/megatron/base/README.md
@@ -1,4 +1,3 @@
 # Base
 
 Base classes and interfaces for Megatron models.
-
diff --git a/dfm/src/megatron/model/README.md b/dfm/src/megatron/model/README.md
@@ -1,4 +1,3 @@
 # Model
 
 Model implementations for DiT and dLLM architectures.
-
diff --git a/examples/README.md b/examples/README.md
@@ -1,4 +1,3 @@
 # Examples
 
 Training and inference examples for dtensor and megatron models
-
diff --git a/examples/dtensor/README.md b/examples/dtensor/README.md
@@ -1,4 +1,3 @@
 # DTensor Models (NeMo Automodel)
 
 Examples using NeMo Automodel with distributed tensor parallelism.
-
diff --git a/examples/dtensor/configs/README.md b/examples/dtensor/configs/README.md
@@ -1,4 +1,3 @@
 # Configs
 
 Configuration files for various Wan model versions.
-
diff --git a/examples/dtensor/scripts/README.md b/examples/dtensor/scripts/README.md
@@ -1,4 +1,3 @@
 # Scripts
 
 Training scripts for pretraining and finetuning.
-
diff --git a/examples/megatron/README.md b/examples/megatron/README.md
@@ -1,4 +1,3 @@
 # Megatron Examples
 
 Recipes and configuration overrides for megatron training.
-
diff --git a/examples/megatron/override_configs/README.md b/examples/megatron/override_configs/README.md
@@ -1,4 +1,3 @@
 # Override Configs
 
 Parallelism configuration overrides for different CP/TP/SP sizes.
-
diff --git a/examples/megatron/recipe/README.md b/examples/megatron/recipe/README.md
@@ -1,4 +1,3 @@
 # Recipe
 
 Training recipes for Wan2.1 pretraining, finetuning, and weight verification.
-
diff --git a/nemo_vfm/diffusion/readme.rst b/nemo_vfm/diffusion/readme.rst
@@ -187,4 +187,4 @@ To launch training on multiple nodes using Slurm
 Citations
 ---------
 
-.. [1] William Peebles and Saining Xie, "Scalable Diffusion Models with Transformers," *arXiv preprint arXiv:2212.09748*, 2022.
+.. [1] William Peebles and Saining Xie, "Scalable Diffusion Models with Transformers," *arXiv preprint arXiv:2212.09748*, 2022.
diff --git a/nemo_vfm/diffusion/vae/readme.rst b/nemo_vfm/diffusion/vae/readme.rst
@@ -124,8 +124,3 @@ We provide a sample training script for launching multi-node training. Simply co
 
    bash nemo/collections/diffusion/vae/train_vae.sh \
    data.path=xxx
-
-
-
-
-
diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/dataverse/av22_tar_index_full.json b/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/dataverse/av22_tar_index_full.json
@@ -1 +1 @@
-{}
+{}
diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/matplotlib/fontlist-v390.json b/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/matplotlib/fontlist-v390.json
@@ -1071,4 +1071,4 @@
     }
   ],
   "__class__": "FontManager"
-}
+}
diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera/AV-V2.2/alpamayo_caption_t5/qwen_t5_tars/trainv2-2-chunk-00/mapping.json b/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera/AV-V2.2/alpamayo_caption_t5/qwen_t5_tars/trainv2-2-chunk-00/mapping.json
@@ -5,4 +5,4 @@
     "1a1c7950-01f0-4e37-9949-dad68e3d1072.camera_rear_tele_30fov": "cosmos1/models/diffusion/nemo/post_training/multicamera/AV-V2.2/alpamayo_caption_t5/qwen_t5_tars/trainv2-2-chunk-00/part_000000/001365.tar",
     "1a1c7950-01f0-4e37-9949-dad68e3d1072.camera_rear_left_70fov": "cosmos1/models/diffusion/nemo/post_training/multicamera/AV-V2.2/alpamayo_caption_t5/qwen_t5_tars/trainv2-2-chunk-00/part_000000/001365.tar",
     "1a1c7950-01f0-4e37-9949-dad68e3d1072.camera_rear_right_70fov": "cosmos1/models/diffusion/nemo/post_training/multicamera/AV-V2.2/AV-V2.2/alpamayo_caption_t5/qwen_t5_tars/trainv2-2-chunk-00/part_000000/001365.tar"    
-}
+}
diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera/AV-V2.2/alpamayo_caption_t5/qwen_t5_tars/trainv2-2-chunk-00/part_000000/001365.json b/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera/AV-V2.2/alpamayo_caption_t5/qwen_t5_tars/trainv2-2-chunk-00/part_000000/001365.json
@@ -1 +1 @@
-{"1a1c7950-01f0-4e37-9949-dad68e3d1072":{"0":["trainv2-2-chunk-00",["The video shows a car driving on a straight road at night. The car's trajectory is consistent, moving forward without any turns. The scene is illuminated by streetlights, and the road appears to be clear of other vehicles or pedestrians. The environment is a well-lit urban area, with no visible weather conditions affecting the visibility. The time of day is nighttime, as indicated by the darkness and artificial lighting."],[2],[130]],"1":["trainv2-2-chunk-00",["The video shows a car driving on a road at night. The car's trajectory indicates a steady forward movement. The scene is illuminated by streetlights, and the road appears to be clear of other vehicles and pedestrians. The weather conditions seem to be clear, with no visible rain or fog. The time of day is nighttime, as indicated by the darkness and artificial lighting."],[2],[130]],"2":["trainv2-2-chunk-00",["The video shows a car driving on a road at night. The car's trajectory indicates it is moving forward. The scene includes streetlights illuminating the road, and there are no other visible vehicles or pedestrians. The environment appears to be a quiet, possibly suburban area with minimal traffic. The weather conditions are not clearly visible, but the lighting suggests it is nighttime."],[2],[130]],"3":["trainv2-2-chunk-00",["The video depicts a car driving on a road at night. The car is moving forward, and the trajectory shows a steady path. The scene is illuminated by streetlights, casting a yellowish glow. Several cars are visible in the distance, and the road appears to be relatively empty. The weather seems clear, with no visible signs of rain or fog. The time of day is nighttime, as indicated by the artificial lighting and the darkness of the surroundings."],[2],[130]]}}
+{"1a1c7950-01f0-4e37-9949-dad68e3d1072":{"0":["trainv2-2-chunk-00",["The video shows a car driving on a straight road at night. The car's trajectory is consistent, moving forward without any turns. The scene is illuminated by streetlights, and the road appears to be clear of other vehicles or pedestrians. The environment is a well-lit urban area, with no visible weather conditions affecting the visibility. The time of day is nighttime, as indicated by the darkness and artificial lighting."],[2],[130]],"1":["trainv2-2-chunk-00",["The video shows a car driving on a road at night. The car's trajectory indicates a steady forward movement. The scene is illuminated by streetlights, and the road appears to be clear of other vehicles and pedestrians. The weather conditions seem to be clear, with no visible rain or fog. The time of day is nighttime, as indicated by the darkness and artificial lighting."],[2],[130]],"2":["trainv2-2-chunk-00",["The video shows a car driving on a road at night. The car's trajectory indicates it is moving forward. The scene includes streetlights illuminating the road, and there are no other visible vehicles or pedestrians. The environment appears to be a quiet, possibly suburban area with minimal traffic. The weather conditions are not clearly visible, but the lighting suggests it is nighttime."],[2],[130]],"3":["trainv2-2-chunk-00",["The video depicts a car driving on a road at night. The car is moving forward, and the trajectory shows a steady path. The scene is illuminated by streetlights, casting a yellowish glow. Several cars are visible in the distance, and the road appears to be relatively empty. The weather seems clear, with no visible signs of rain or fog. The time of day is nighttime, as indicated by the artificial lighting and the darkness of the surroundings."],[2],[130]]}}

Original file line number	Diff line number	Diff line change
`@@ -27,4 +27,4 @@ If you find our code useful, please consider citing the following papers:`
`27`	`27`	`journal={arXiv preprint arXiv:2501.03575},`
`28`	`28`	`year={2025}`
`29`	`29`	`}`
`30`		-```
	`30`	+```
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,3 @@`
`1`	`1`	`# Automodel`
`2`	`2`
`3`	`3`	`Models implemented using Dtensors (NeMo Automodel)`
`4`		`-`
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,3 @@`
`1`	`1`	`# MBridge`
`2`	`2`
`3`	`3`	`Megatron models including DiT and dLLM implementations.`
`4`		`-`
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,3 @@`
`1`	`1`	`# Base`
`2`	`2`
`3`	`3`	`Base classes and interfaces for Megatron models.`
`4`		`-`
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,3 @@`
`1`	`1`	`# Model`
`2`	`2`
`3`	`3`	`Model implementations for DiT and dLLM architectures.`
`4`		`-`
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,3 @@`
`1`	`1`	`# Examples`
`2`	`2`
`3`	`3`	`Training and inference examples for dtensor and megatron models`
`4`		`-`
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,3 @@`
`1`	`1`	`# DTensor Models (NeMo Automodel)`
`2`	`2`
`3`	`3`	`Examples using NeMo Automodel with distributed tensor parallelism.`
`4`		`-`