diff --git a/scripts/data.py b/scripts/data.py index 0a72cbbfc8af..72a6bbfc64c0 100644 --- a/scripts/data.py +++ b/scripts/data.py @@ -1,6 +1,6 @@ import os from pathlib import Path - +SRC_DIR = Path(__file__).parent def load_prompt(): try: @@ -9,7 +9,7 @@ def load_prompt(): data_dir = file_dir / "data" prompt_file = data_dir / "prompt.txt" # Load the promt from data/prompt.txt - with open(prompt_file, "r") as prompt_file: + with open(SRC_DIR/ "data/prompt.txt", "r") as prompt_file: prompt = prompt_file.read() return prompt