a

eisenzopf · May 29, 2024 · af81f45 · af81f45
1 parent 77ca908
commit af81f45
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/llm_eval/handler.py b/llm_eval/handler.py
@@ -74,7 +74,7 @@ def post_process_output(self, prompt, output):
         if self.current_model == 'meta-llama/Llama-2-7b-chat-hf':
             output = output[len(prompt)-1:]
         pattern = re.compile(r'\{\s*"(.+?)"\s*:\s*"(.+?)"\s*\}')
-        match = re.findall(pattern, "target_text")[-1]
+        match = re.findall(pattern, output)[-1]
         return {match.group(1): match.group(2)} if match else output
 
     def prepare_output(self):

diff --git a/pyproject.toml b/pyproject.toml
@@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "llm-eval"
-version = "0.5.10"
+version = "0.5.11"
 authors = [
     {name = "Jonathan Eisenzopf", email = "[email protected]"},
 ]