Working to simplify output.

eisenzopf · May 28, 2024 · 7558671 · 7558671
1 parent cec8d66
commit 7558671
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/llm_eval/handler.py b/llm_eval/handler.py
@@ -108,6 +108,7 @@ def process_dataset(self):
                     if col.endswith('.input'):
                         output_col = col.replace('.input', '.output')
                         prompt, output = self.generate_output(row[col])
+                        output = self.post_process_output(output)
                         df.at[index, output_col] = output
             self.unload_model(model_name)
         return df

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