From d8a5f185b8ca5c04d78e993948b12001096843a8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Daniel=20Grie=C3=9Fhaber?= <griesshaber@hdm-stuttgart.de>
Date: Fri, 9 Aug 2024 09:45:30 +0200
Subject: [PATCH] log and return correct prompts in get_initial_prompts
---
evoprompt/optimization.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/evoprompt/optimization.py b/evoprompt/optimization.py
index e27ce3d..6aa71e7 100644
--- a/evoprompt/optimization.py
+++ b/evoprompt/optimization.py
@@ -129,13 +129,15 @@ class PromptOptimization:
initial_population += paraphrases
logger.info(
"Paraphrased prompt '%s': %s.",
- self.task.base_prompts[0].replace("\r", "\\r").replace("\n", "\\n"),
+ top_prompts[promptindex_to_paraphrase]
+ .replace("\r", "\\r")
+ .replace("\n", "\\n"),
paraphrases,
)
promptindex_to_paraphrase += 1
promptindex_to_paraphrase %= len(top_prompts)
- return self.task.base_prompts + paraphrases
+ return initial_population
def add_prompt(
self,
--
GitLab