Update app/openai_chat_completion.py

app/openai_chat_completion.py

@@ -42,7 +42,7 @@ class OpenAIChatCompletions:
 
 
     # function to use test data to predict completions
-    def predict_jsonl(self, path_or_buf='data-cleaning-llm/data/cookies_test.jsonl', n_samples=None, n_shot=None):
+    def predict_jsonl(self, path_or_buf='data/cookies_test.jsonl', n_samples=None, n_shot=None):
         jsonObj = pd.read_json(path_or_buf=path_or_buf, lines=True)
         if n_samples is not None:
             jsonObj = jsonObj.sample(n_samples, random_state=42)
@@ -71,6 +71,6 @@ class OpenAIChatCompletions:
 
     # a method that samples n rows from a jsonl file, returning a pandas dataframe
     @staticmethod
-    def _sample_jsonl(path_or_buf='data-cleaning-llm/data/cookies_train.jsonl', n_samples=5):
+    def _sample_jsonl(path_or_buf='data/cookies_train.jsonl', n_samples=5):
         jsonObj = pd.read_json(path_or_buf=path_or_buf, lines=True)
         return jsonObj.sample(n_samples, random_state=42)