fmt

refuel-ai · Oct 23, 2024 · 951698a · 951698a
1 parent 7b69613
commit 951698a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/autolabel/tasks/attribute_extraction.py b/src/autolabel/tasks/attribute_extraction.py
@@ -363,9 +363,9 @@ def parse_llm_response(
                                 original_attr_labels,
                             )
                         )
-                        llm_label[attribute["name"]] = (
-                            self.config.label_separator().join(filtered_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}"