From 0903ec5d58f96b7379cd4600b5a5fb5b981deb30 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Andreas=20K=C3=B6pf?= Date: Sat, 28 Jan 2023 00:03:52 +0100 Subject: [PATCH] fix quility label addition --- backend/oasst_backend/tree_manager.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/backend/oasst_backend/tree_manager.py b/backend/oasst_backend/tree_manager.py index d6e5bcfc..782b9af8 100644 --- a/backend/oasst_backend/tree_manager.py +++ b/backend/oasst_backend/tree_manager.py @@ -377,7 +377,7 @@ class TreeManager: label_mode = protocol_schema.LabelTaskMode.simple label_disposition = protocol_schema.LabelTaskDisposition.spam valid_labels = list(self.cfg.mandatory_labels_assistant_reply) - if protocol_schema.LabelTaskDisposition.quality not in valid_labels: + if protocol_schema.TextLabel.quality not in valid_labels: valid_labels.append(protocol_schema.TextLabel.quality) logger.info(f"Generating a LabelAssistantReplyTask. ({label_mode=:s})") @@ -401,7 +401,7 @@ class TreeManager: label_mode = protocol_schema.LabelTaskMode.simple label_disposition = protocol_schema.LabelTaskDisposition.spam valid_labels = list(self.cfg.mandatory_labels_prompter_reply) - if protocol_schema.LabelTaskDisposition.quality not in valid_labels: + if protocol_schema.TextLabel.quality not in valid_labels: valid_labels.append(protocol_schema.TextLabel.quality) logger.info(f"Generating a LabelPrompterReplyTask. ({label_mode=:s})")