Skip to content

Commit 727d66d

Browse files
committed
type hint
Signed-off-by: wangli <wangli858794774@gmail.com>
1 parent 06b0a75 commit 727d66d

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

tests/e2e/singlecard/test_guided_decoding.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,7 @@ def test_guided_json_completion(guided_decoding_backend: str,
108108
"max_tokens": 500,
109109
})
110110

111-
runner_kwargs = {
111+
runner_kwargs: Dict[str, Any] = {
112112
"seed": 0,
113113
}
114114
if vllm_version_is("0.10.2"):
@@ -154,7 +154,7 @@ def test_guided_regex(guided_decoding_backend: str, sample_regex):
154154
"top_p": 0.95,
155155
})
156156

157-
runner_kwargs = {
157+
runner_kwargs: Dict[str, Any] = {
158158
"seed": 0,
159159
}
160160
if vllm_version_is("0.10.2"):
@@ -164,8 +164,7 @@ def test_guided_regex(guided_decoding_backend: str, sample_regex):
164164
"backend": guided_decoding_backend
165165
}
166166

167-
with VllmRunner(MODEL_NAME,
168-
**runner_kwargs) as vllm_model: # type: ignore[arg-type]
167+
with VllmRunner(MODEL_NAME, **runner_kwargs) as vllm_model:
169168
prompts = [
170169
f"Give an example IPv4 address with this regex: {sample_regex}"
171170
] * 2

0 commit comments

Comments
 (0)