Skip to content

Commit 5a5ef91

Browse files
committed
Polishing
Signed-off-by: wang.yuqi <noooop@126.com>
1 parent c6d4cb7 commit 5a5ef91

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/models/language/pooling/test_token_classification.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
pytest.param("boltuix/NeuroBERT-NER", ),
1414
],
1515
)
16+
# The float32 is required for this tiny model to pass the test.
1617
@pytest.mark.parametrize("dtype", ["float"])
1718
@torch.inference_mode
1819
def test_models(
@@ -22,8 +23,6 @@ def test_models(
2223
model: str,
2324
dtype: str,
2425
) -> None:
25-
# The float32 is required for this tiny model to pass the test.
26-
2726
with vllm_runner(model,
2827
max_model_len=None,
2928
dtype=dtype,

0 commit comments

Comments
 (0)