[Refactor]: clean up diffusion scheduler interfaces (#5216)

Signed-off-by: Huang, Zeyu <11222265+fhfuih@users.noreply.github.com>
Signed-off-by: Zeyu Huang | 黃澤宇 <11222265+fhfuih@users.noreply.github.com>
Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>
Co-authored-by: Cursor <cursoragent@cursor.com>

tests/ar_diffusion/test_engine_routing.py

@@ -136,8 +136,8 @@ def test_runner_rejects_multi_sequence_config():
         ARDiffusionModelRunner._preallocate_kv_cache(fake)
 
 
-def test_execute_model_accepts_kv_prefetch_jobs_kwarg():
-    """Regression: the worker passes ``kv_prefetch_jobs`` to every runner
+def test_execute_model_accepts_kv_prefetch_job_kwarg():
+    """Regression: the worker passes ``kv_prefetch_job`` to every runner
     (added with the KV prefetch path, #4448); the AR-Diffusion override must
     accept and forward it or DreamZero crashes on the warm-up dummy run."""
     import inspect
@@ -147,4 +147,4 @@ def test_execute_model_accepts_kv_prefetch_jobs_kwarg():
 
     for cls in (DiffusionModelRunner, ARDiffusionModelRunner):
         params = inspect.signature(cls.execute_model).parameters
-        assert "kv_prefetch_jobs" in params, cls
+        assert "kv_prefetch_job" in params, cls