[Testing] Fix core tests (#3224)

2024-03-06 01:04:23 -08:00
parent 24aecf421a
commit a33ce60c66
4 changed files with 36 additions and 23 deletions
--- a/tests/core/utils.py
+++ b/tests/core/utils.py
@@ -18,7 +18,7 @@ def create_dummy_prompt(
    prompt_str = " ".join([str(t) for t in prompt_tokens])
    prompt = Sequence(int(request_id), prompt_str, prompt_tokens, block_size)
    seq_group = SequenceGroup(request_id, [prompt], SamplingParams(),
-                              time.time(), None, None)
+                              time.time(), None)

    return prompt, seq_group