From 4aaafdd289f57a82513a7742155e4f1b796c8bdc 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 16c04e01d703..8d6afdbd0035 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -8,8 +8,9 @@ from transformers import AutoModelForCausalLM 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