wassname/Open-Assistant
1
0
Fork 0
mirror of https://github.com/wassname/Open-Assistant.git synced 2026-07-18 12:10:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix valid_labels/mandatory labels mixup

Browse Source
This commit is contained in:
Andreas Köpf
2023-01-30 00:26:09 +01:00
parent e59a4393dc
commit e61c1c0ca2
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/oasst_backend/tree_manager.py
+3 -3
View File
@@ -375,7 +375,7 @@ class TreeManager:
):
label_mode = protocol_schema.LabelTaskMode.simple
label_disposition = protocol_schema.LabelTaskDisposition.spam
valid_labels = list(self.cfg.mandatory_labels_assistant_reply)
valid_labels = self.cfg.mandatory_labels_assistant_reply.copy()
if protocol_schema.TextLabel.lang_mismatch not in valid_labels:
valid_labels.append(protocol_schema.TextLabel.lang_mismatch)
if protocol_schema.TextLabel.quality not in valid_labels:
@@ -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)
valid_labels = self.cfg.mandatory_labels_prompter_reply.copy()
if protocol_schema.TextLabel.lang_mismatch not in valid_labels:
valid_labels.append(protocol_schema.TextLabel.lang_mismatch)
if protocol_schema.TextLabel.quality not in valid_labels:
@@ -484,7 +484,7 @@ class TreeManager:
valid_labels = self.cfg.labels_initial_prompt
if random.random() > self.cfg.p_full_labeling_review_prompt:
valid_labels = self.cfg.mandatory_labels_initial_prompt
valid_labels = self.cfg.mandatory_labels_initial_prompt.copy()
label_mode = protocol_schema.LabelTaskMode.simple
label_disposition = protocol_schema.LabelTaskDisposition.spam
if protocol_schema.TextLabel.lang_mismatch not in valid_labels: