Skip to content

Commit 02d4b85

Browse files
Use kwargs for long lists of EngineCoreRequest arguments in tests and fix extra kwargs (#24987)
Signed-off-by: Andrew Sansom <andrew@protopia.ai>
1 parent 86daa87 commit 02d4b85

File tree

2 files changed

+16
-18
lines changed

2 files changed

+16
-18
lines changed

tests/detokenizer/test_min_tokens.py

Lines changed: 8 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -31,16 +31,14 @@ def test_min_tokens_with_stop(min_tokens: int, stop: str, truth: str):
3131
stop=stop,
3232
min_tokens=min_tokens,
3333
)
34-
request = EngineCoreRequest("",
35-
prompt_token_ids,
36-
None,
37-
None,
38-
None,
39-
params,
40-
None,
41-
None,
42-
0.0,
43-
None,
34+
request = EngineCoreRequest(request_id="",
35+
prompt_token_ids=prompt_token_ids,
36+
mm_features=None,
37+
sampling_params=params,
38+
pooling_params=None,
39+
eos_token_id=None,
40+
arrival_time=0.0,
41+
lora_request=None,
4442
cache_salt=None,
4543
data_parallel_rank=None)
4644

tests/tokenization/test_detokenize.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -61,14 +61,14 @@ def _run_incremental_decode(tokenizer,
6161
skip_special_tokens=skip_special_tokens,
6262
spaces_between_special_tokens=spaces_between_special_tokens,
6363
)
64-
request = EngineCoreRequest("",
65-
prompt_token_ids,
66-
None,
67-
params,
68-
None,
69-
None,
70-
0.0,
71-
None,
64+
request = EngineCoreRequest(request_id="",
65+
prompt_token_ids=prompt_token_ids,
66+
mm_features=None,
67+
sampling_params=params,
68+
pooling_params=None,
69+
eos_token_id=None,
70+
arrival_time=0.0,
71+
lora_request=None,
7272
cache_salt=None,
7373
data_parallel_rank=None)
7474

0 commit comments

Comments
 (0)