File tree Expand file tree Collapse file tree 1 file changed +5
-4
lines changed
Expand file tree Collapse file tree 1 file changed +5
-4
lines changed Original file line number Diff line number Diff line change 88from vllm import LLM , SamplingParams
99from vllm .transformers_utils .tokenizer import get_tokenizer
1010
11- _TEST_PROMPTS = ["prompts/example.txt" ]
12- _LONG_PROMPTS = ["prompts/summary.txt" ]
11+ _TEST_DIR = os .path .dirname (__file__ )
12+ _TEST_PROMPTS = [os .path .join (_TEST_DIR , "prompts" , "example.txt" )]
13+ _LONG_PROMPTS = [os .path .join (_TEST_DIR , "prompts" , "summary.txt" )]
1314
1415
1516def _read_prompts (filename : str ) -> str :
@@ -24,15 +25,15 @@ def _read_prompts(filename: str) -> str:
2425def example_prompts () -> List [str ]:
2526 prompts = []
2627 for filename in _TEST_PROMPTS :
27- prompts += _read_prompts (os . path . join ( "tests" , filename ) )
28+ prompts += _read_prompts (filename )
2829 return prompts
2930
3031
3132@pytest .fixture
3233def example_long_prompts () -> List [str ]:
3334 prompts = []
3435 for filename in _LONG_PROMPTS :
35- prompts += _read_prompts (os . path . join ( "tests" , filename ) )
36+ prompts += _read_prompts (filename )
3637 return prompts
3738
3839
You can’t perform that action at this time.
0 commit comments