From 725dcdcec90057b9a302b1ee21c6f3f2503a73c6 Mon Sep 17 00:00:00 2001
From: deep1 <>
Date: Mon, 11 Sep 2023 06:45:05 +0800
Subject: [PATCH] fix for amazon_polarity where answers have a newline first

---
 src/prompts/prompt_loading.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/prompts/prompt_loading.py b/src/prompts/prompt_loading.py
index 675f332..aa3cd21 100644
--- a/src/prompts/prompt_loading.py
+++ b/src/prompts/prompt_loading.py
@@ -226,7 +226,7 @@ def _convert_to_prompts(
                     fewshot_examples = next(fewshot_iter)
                     if instructed_to_lie: fewshot_examples = [{**e, 'label': ~e['label']} for e in fewshot_examples]
                     fewshot_texts = [
-                        dict(user=q, response=a) for q, a in map(template.apply, fewshot_examples)
+                        dict(user=q, response=a.strip()) for q, a in map(template.apply, fewshot_examples)
                     ]
                     prompt_parts = fewshot_texts + prompt_parts