From 5e5d7541d21a2d6311c4818ebd0346898b14b3bb Mon Sep 17 00:00:00 2001
From: Maximilian Schmidt <maximilian.schmidt@ims.uni-stuttgart.de>
Date: Wed, 28 Aug 2024 09:40:25 +0200
Subject: [PATCH] Log invalid answer as debug to not pollute terminal
---
evoprompt/task/text_classification.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/evoprompt/task/text_classification.py b/evoprompt/task/text_classification.py
index 1e19766..2c40725 100644
--- a/evoprompt/task/text_classification.py
+++ b/evoprompt/task/text_classification.py
@@ -49,7 +49,7 @@ class TextClassification(Task):
class_mapping = self._get_label_mapping()
response = response.lower()
if response not in class_mapping:
- logger.warning(f"Invalid answer: {response}")
+ logger.debug(f"Invalid answer: {response}")
return "failed"
prediction = class_mapping[response]
classification_result = "incorrect" if prediction != gold_label else "correct"
--
GitLab