From 3b07c6b17356b342804b9fba0d85ec331c06a14c Mon Sep 17 00:00:00 2001 From: Zhuohan Li Date: Wed, 27 Dec 2023 02:37:21 +0800 Subject: [PATCH] [BUGFIX] Fix the path of test prompts (#2273) --- tests/conftest.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py index 16c04e01d703c..8d6afdbd00358 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -8,8 +8,9 @@ from vllm import LLM, SamplingParams from vllm.transformers_utils.tokenizer import get_tokenizer -_TEST_PROMPTS = ["prompts/example.txt"] -_LONG_PROMPTS = ["prompts/summary.txt"] +_TEST_DIR = os.path.dirname(__file__) +_TEST_PROMPTS = [os.path.join(_TEST_DIR, "prompts", "example.txt")] +_LONG_PROMPTS = [os.path.join(_TEST_DIR, "prompts", "summary.txt")] def _read_prompts(filename: str) -> str: @@ -24,7 +25,7 @@ def _read_prompts(filename: str) -> str: def example_prompts() -> List[str]: prompts = [] for filename in _TEST_PROMPTS: - prompts += _read_prompts(os.path.join("tests", filename)) + prompts += _read_prompts(filename) return prompts @@ -32,7 +33,7 @@ def example_prompts() -> List[str]: def example_long_prompts() -> List[str]: prompts = [] for filename in _LONG_PROMPTS: - prompts += _read_prompts(os.path.join("tests", filename)) + prompts += _read_prompts(filename) return prompts