vllm

1
"""Tests for the SamplingParams class.
2
"""
3
from vllm import SamplingParams
4

5

6
def test_max_tokens_none():
7
    """max_tokens=None should be allowed"""
8
    SamplingParams(temperature=0.01, top_p=0.1, max_tokens=None)
9

10

11
if __name__ == "__main__":
12
    import pytest
13
    pytest.main([__file__])
14