From e8c8820c63da85f394439f60605f4569efe57e33 Mon Sep 17 00:00:00 2001 From: Kirill Vasilev Date: Mon, 3 Aug 2026 12:09:15 -0700 Subject: [PATCH] test: update orpheus modular test to the decode_finished contract The prefill-transition test predates the dynamic loop-stop: the model no longer seeds audio_token_count/decode_finished kwargs at the prefill->decode transition (decode_finished is set only at loop exit). Assert the current contract instead; drop the dead audio_token_count input and an unused helper. --- test/modular/test_orpheus_model.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/test/modular/test_orpheus_model.py b/test/modular/test_orpheus_model.py index 28071b853..d4afe42e3 100644 --- a/test/modular/test_orpheus_model.py +++ b/test/modular/test_orpheus_model.py @@ -14,10 +14,6 @@ def _make_model() -> OrpheusModel: return model -def _audio_token_for_pos(pos: int, code: int = 1) -> int: - return 10 + (pos * 4096) + code - - def test_orpheus_prefill_transitions_to_decode(): model = _make_model() metadata = CurrentForwardConductorMetadata( @@ -35,8 +31,8 @@ def test_orpheus_prefill_transitions_to_decode(): assert result.full_metadata.graph_walk == "decode" assert result.step_metadata["is_prefill"] is False - assert result.full_metadata.kwargs["audio_token_count"] == 0 - assert result.full_metadata.kwargs["decode_finished"] is False + assert result.request_done is False + assert "decode_finished" not in result.full_metadata.kwargs def test_orpheus_decode_eos_marks_done(): @@ -47,7 +43,6 @@ def test_orpheus_decode_eos_marks_done(): graph_walk="decode", is_prefill=False, kwargs={ - "audio_token_count": 0, "decode_finished": False, }, )