Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
c81259d3
Unverified
Commit
c81259d3
authored
Jul 19, 2025
by
Rabi Mishra
Committed by
GitHub
Jul 19, 2025
Browse files
Fix/remove some broken model executor tests (#21224)
Signed-off-by:
Rabi Mishra
<
ramishra@redhat.com
>
parent
e3a0e43d
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
3 additions
and
16 deletions
+3
-16
tests/model_executor/test_guided_processors.py
tests/model_executor/test_guided_processors.py
+0
-13
tests/model_executor/test_model_load_with_params.py
tests/model_executor/test_model_load_with_params.py
+3
-3
No files found.
tests/model_executor/test_guided_processors.py
View file @
c81259d3
...
@@ -189,19 +189,6 @@ def test_multiple_guided_options_not_allowed(sample_json_schema, sample_regex):
...
@@ -189,19 +189,6 @@ def test_multiple_guided_options_not_allowed(sample_json_schema, sample_regex):
GuidedDecodingParams
(
json
=
sample_json_schema
,
grammar
=
"test grammar"
)
GuidedDecodingParams
(
json
=
sample_json_schema
,
grammar
=
"test grammar"
)
def
test_guided_decoding_backend_options
():
"""Test backend-specific options"""
with
pytest
.
warns
(
DeprecationWarning
):
guided_decoding_params
=
GuidedDecodingParams
(
backend
=
"xgrammar:no-fallback,disable-any-whitespace,no-additional-properties"
)
assert
guided_decoding_params
.
backend
==
"xgrammar"
assert
guided_decoding_params
.
disable_fallback
assert
guided_decoding_params
.
disable_any_whitespace
assert
guided_decoding_params
.
disable_additional_properties
def
test_pickle_xgrammar_tokenizer_data
():
def
test_pickle_xgrammar_tokenizer_data
():
try
:
try
:
import
xgrammar
as
xgr
import
xgrammar
as
xgr
...
...
tests/model_executor/test_model_load_with_params.py
View file @
c81259d3
...
@@ -49,7 +49,7 @@ def test_model_loading_with_params(vllm_runner):
...
@@ -49,7 +49,7 @@ def test_model_loading_with_params(vllm_runner):
def
check_model
(
model
):
def
check_model
(
model
):
assert
isinstance
(
model
,
BertEmbeddingModel
)
assert
isinstance
(
model
,
BertEmbeddingModel
)
assert
isinstance
(
model
.
_
pooler
,
CLSPool
)
assert
isinstance
(
model
.
pooler
.
pooling
,
CLSPool
)
vllm_model
.
apply_model
(
check_model
)
vllm_model
.
apply_model
(
check_model
)
...
@@ -87,7 +87,7 @@ def test_roberta_model_loading_with_params(vllm_runner):
...
@@ -87,7 +87,7 @@ def test_roberta_model_loading_with_params(vllm_runner):
def
check_model
(
model
):
def
check_model
(
model
):
assert
isinstance
(
model
,
RobertaEmbeddingModel
)
assert
isinstance
(
model
,
RobertaEmbeddingModel
)
assert
isinstance
(
model
.
_
pooler
,
MeanPool
)
assert
isinstance
(
model
.
pooler
.
pooling
,
MeanPool
)
vllm_model
.
apply_model
(
check_model
)
vllm_model
.
apply_model
(
check_model
)
...
@@ -114,7 +114,7 @@ def test_facebook_roberta_model_loading_with_params(vllm_runner):
...
@@ -114,7 +114,7 @@ def test_facebook_roberta_model_loading_with_params(vllm_runner):
def
check_model
(
model
):
def
check_model
(
model
):
assert
isinstance
(
model
,
RobertaEmbeddingModel
)
assert
isinstance
(
model
,
RobertaEmbeddingModel
)
assert
not
hasattr
(
model
,
"lm_head"
)
assert
not
hasattr
(
model
,
"lm_head"
)
assert
isinstance
(
model
.
_
pooler
,
CLSPool
)
assert
isinstance
(
model
.
pooler
.
pooling
,
CLSPool
)
vllm_model
.
apply_model
(
check_model
)
vllm_model
.
apply_model
(
check_model
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment