Merge pull request Significant-Gravitas#50 from tomviner/fix-prompt-path

don't depend on the path script is run from
Sneaking · Apr 4, 2023 · 37f8466 · 37f8466
2 parents 3088050 + 4d4ff65
commit 37f8466
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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