[Fix] Align diffusion proc test mock with current output fields (#2584)

Signed-off-by: Jinheng Li <ahengljh@gmail.com>
This commit is contained in:
Jinheng authored and GitHub committed 2026-04-08 17:45:10 +08:00
1 parent fb3c6bd9b1
commit aefa2ee45c
1 file changed
+10
@@ -24,19 +24,29 @@ def test_process_batch_request_preserves_parent_request_id_and_kv_sender_info():
SimpleNamespace(
images=["img-1"],
_multimodal_output={},
_custom_output={},
metrics={},
stage_durations={},
peak_memory_mb=0.0,
latents=None,
trajectory_latents=None,
trajectory_timesteps=None,
trajectory_log_probs=None,
trajectory_decoded=None,
final_output_type="image",
),
SimpleNamespace(
images=["img-2"],
_multimodal_output={},
_custom_output={},
metrics={},
stage_durations={},
peak_memory_mb=0.0,
latents=None,
trajectory_latents=None,
trajectory_timesteps=None,
trajectory_log_probs=None,
trajectory_decoded=None,
final_output_type="image",
),
]