Skip to content

test: update stale orpheus modular test to the decode_finished contract #205

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
vasilevklart wants to merge 1 commit into
base: main
Choose a base branch
from
Open

test: update stale orpheus modular test to the decode_finished contract #205

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 2 additions & 7 deletions test/modular/test_orpheus_model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand All @@ -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():
Expand All @@ -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,
},
)
Expand Down
Loading