[test]: add basic unit test for the generic pipeline runner

Author: techverve

fastvideo/tests/training/test_runner.py

@@ -0,0 +1,18 @@
+import subprocess
+import sys
+
+def test_runner_help():
+    # Run the runner with --help to ensure it parses arguments correctly
+    # and doesn't have any syntax errors or missing imports.
+    result = subprocess.run(
+        [sys.executable, "-m", "fastvideo.training.runner", "--help"],
+        capture_output=True,
+        text=True
+    )
+    
+    # Check if the process completed successfully
+    assert result.returncode == 0, f"Runner failed with exit code {result.returncode}.\nStderr: {result.stderr}"
+    
+    # Check if help output is printed
+    assert "--pipeline_class" in result.stdout
+    assert "--pipeline_module" in result.stdout

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)