[style]: apply pre-commit formatting

Author: techverve

fastvideo/training/cosmos2_5_training_pipeline.py

@@ -132,4 +132,3 @@ def _build_input_kwargs(self, training_batch: TrainingBatch) -> TrainingBatch:
 # ---------------------------------------------------------------------------
 # Entry point (mirrors wan_training_pipeline.py)
 # ---------------------------------------------------------------------------
-

fastvideo/training/ltx2_training_pipeline.py

@@ -1,6 +1,5 @@
 # SPDX-License-Identifier: Apache-2.0
 import os
-import sys
 from copy import deepcopy
 from pathlib import Path
 import torch
@@ -473,4 +472,3 @@ def _clip_grad_norm(self, training_batch: TrainingBatch) -> TrainingBatch:
             grad_norm = 0.0
         training_batch.grad_norm = grad_norm
         return training_batch
-

fastvideo/training/matrixgame2_ar_diffusion_pipeline.py

@@ -1,6 +1,5 @@
 # SPDX-License-Identifier: Apache-2.0
 
-import sys
 from copy import deepcopy
 from typing import Any, cast
 
@@ -426,4 +425,3 @@ def _prepare_validation_batch(
             batch.mouse_cond = mouse_cond
 
         return batch
-

fastvideo/training/matrixgame2_self_forcing_distillation_pipeline.py

@@ -1,5 +1,4 @@
 # SPDX-License-Identifier: Apache-2.0
-import sys
 from collections.abc import Iterable
 from copy import deepcopy
 from typing import Any, cast
@@ -874,4 +873,3 @@ def _prepare_validation_batch(self, sampling_param: SamplingParam, training_args
             batch.mouse_cond = mouse_cond
 
         return batch
-

fastvideo/training/matrixgame2_training_pipeline.py

@@ -1,5 +1,4 @@
 # SPDX-License-Identifier: Apache-2.0
-import sys
 from copy import deepcopy
 from typing import Any
 
@@ -193,4 +192,3 @@ def _prepare_validation_batch(self, sampling_param: SamplingParam, training_args
             batch.mouse_cond = mouse_cond
 
         return batch
-

fastvideo/training/ode_causal_pipeline.py

@@ -1,5 +1,4 @@
 # SPDX-License-Identifier: Apache-2.0
-import sys
 from copy import deepcopy
 from typing import Any, cast
 
@@ -308,4 +307,3 @@ def visualize_intermediate_latents(self, training_batch: TrainingBatch, training
 
         if self.global_rank == 0 and tracker_loss_dict:
             self.tracker.log_artifacts(tracker_loss_dict, step)
-

fastvideo/training/runner.py

@@ -22,8 +22,14 @@ def main(args) -> None:
 
 if __name__ == "__main__":
     parser = FlexibleArgumentParser()
-    parser.add_argument("--pipeline_class", type=str, required=True, help="Name of the pipeline class to run, e.g., WanTrainingPipeline")
-    parser.add_argument("--pipeline_module", type=str, required=True, help="Module containing the pipeline class, e.g., fastvideo.training.wan_training_pipeline")
+    parser.add_argument("--pipeline_class",
+                        type=str,
+                        required=True,
+                        help="Name of the pipeline class to run, e.g., WanTrainingPipeline")
+    parser.add_argument("--pipeline_module",
+                        type=str,
+                        required=True,
+                        help="Module containing the pipeline class, e.g., fastvideo.training.wan_training_pipeline")
     parser = TrainingArgs.add_cli_args(parser)
     parser = FastVideoArgs.add_cli_args(parser)
 

fastvideo/training/wan_distillation_pipeline.py

@@ -1,5 +1,4 @@
 # SPDX-License-Identifier: Apache-2.0
-import sys
 from copy import deepcopy
 
 from fastvideo.fastvideo_args import FastVideoArgs, TrainingArgs
@@ -51,4 +50,3 @@ def initialize_validation_pipeline(self, training_args: TrainingArgs):
             dit_cpu_offload=True)
 
         self.validation_pipeline = validation_pipeline
-

fastvideo/training/wan_i2v_distillation_pipeline.py

@@ -1,5 +1,4 @@
 # SPDX-License-Identifier: Apache-2.0
-import sys
 from copy import deepcopy
 from typing import Any
 
@@ -173,4 +172,3 @@ def _build_distill_input_kwargs(self, noise_input: torch.Tensor, timestep: torch
         training_batch.noise_latents = noise_input
 
         return training_batch
-

fastvideo/training/wan_i2v_training_pipeline.py

@@ -1,5 +1,4 @@
 # SPDX-License-Identifier: Apache-2.0
-import sys
 from copy import deepcopy
 from typing import Any
 
@@ -164,4 +163,3 @@ def _prepare_validation_batch(self, sampling_param: SamplingParam, training_args
         )
 
         return batch
-