revert change for multilabel check

refuel-ai · Oct 23, 2024 · d419584 · d419584
1 parent 9adf432
commit d419584
Showing 1 changed file with 0 additions and 16 deletions.
diff --git a/src/autolabel/tasks/attribute_extraction.py b/src/autolabel/tasks/attribute_extraction.py
@@ -353,22 +353,6 @@ def parse_llm_response(
                                 f"Attribute {attr_label} from the LLM response {llm_label} is not in the labels list"
                             )
                             llm_label.pop(attribute["name"], None)
-                    elif attr_type == TaskType.MULTILABEL_CLASSIFICATION:
-                        original_attr_labels = attr_label.split(
-                            self.config.label_separator()
-                        )
-                        filtered_attr_labels = list(
-                            filter(lambda x: x in attr_options, original_attr_labels)
-                        )
-                        llm_label[
-                            attribute["name"]
-                        ] = self.config.label_separator().join(filtered_attr_labels)
-                        if len(filtered_attr_labels) != len(original_attr_labels):
-                            logger.warning(
-                                f"Attribute {attr_label} from the LLM response {llm_label} is not in the labels list. Filtered list: {filtered_attr_labels}"
-                            )
-                        if len(filtered_attr_labels) == 0:
-                            llm_label.pop(attribute["name"], None)
         return LLMAnnotation(
             curr_sample=pickle.dumps(curr_sample),
             successfully_labeled=successfully_labeled,