From e6c84fe2fdaa3b91aca1230dff48423a1fd62d9b Mon Sep 17 00:00:00 2001 From: Mac Lee Date: Mon, 29 Jun 2026 07:57:02 +0000 Subject: [PATCH] [bugfix]: update batching schema test fixtures (#1453) --- docs/design/inference_schema_parity_inventory.yaml | 5 +++++ fastvideo/tests/api/test_parser.py | 7 +++++++ 2 files changed, 12 insertions(+) diff --git a/docs/design/inference_schema_parity_inventory.yaml b/docs/design/inference_schema_parity_inventory.yaml index 55fc5f158..3b1b3dd6c 100644 --- a/docs/design/inference_schema_parity_inventory.yaml +++ b/docs/design/inference_schema_parity_inventory.yaml @@ -41,6 +41,11 @@ surfaces: disable_autocast: generator.engine.disable_autocast enable_stage_verification: generator.engine.enable_stage_verification prompt_txt: request.inputs.prompt_path + batching_mode: generator.engine.batching.mode + batching_max_size: generator.engine.batching.max_size + batching_delay_ms: generator.engine.batching.delay_ms + batching_config: generator.engine.batching.config_path + enable_batching_metrics: generator.engine.batching.enable_metrics override_text_encoder_safetensors: generator.pipeline.components.text_encoder_weights override_text_encoder_quant: generator.engine.quantization.text_encoder_quant transformer_quant: generator.engine.quantization.transformer_quant diff --git a/fastvideo/tests/api/test_parser.py b/fastvideo/tests/api/test_parser.py index 91a2bbfe4..8c6aeecc7 100644 --- a/fastvideo/tests/api/test_parser.py +++ b/fastvideo/tests/api/test_parser.py @@ -130,6 +130,13 @@ def test_load_run_config_supports_yaml_roundtrip(tmp_path) -> None: "use_fsdp_inference": False, "disable_autocast": False, "quantization": None, + "batching": { + "mode": "disabled", + "max_size": 1, + "delay_ms": 0.0, + "config_path": None, + "enable_metrics": False, + }, }, "pipeline": { "workload_type": None,