Files
vllm/tests/samplers/test_beam_search.py