Move serialization outside autolabel (#929)

refuel-ai · Nov 1, 2024 · 3561097 · 3561097
1 parent fca9d03
commit 3561097
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/autolabel/tasks/attribute_extraction.py b/src/autolabel/tasks/attribute_extraction.py
@@ -312,7 +312,7 @@ def parse_llm_response(
             llm_label = {}
             for k, v in json5.loads(completion_text).items():
                 if isinstance(v, list) or isinstance(v, dict):
-                    llm_label[k] = json.dumps(v)
+                    llm_label[k] = v
                 else:
                     llm_label[k] = str(v)
             successfully_labeled = True
@@ -327,7 +327,7 @@ def parse_llm_response(
                     response.text[json_start : json_end + 1]
                 ).items():
                     if isinstance(v, list) or isinstance(v, dict):
-                        llm_label[k] = json.dumps(v)
+                        llm_label[k] = v
                     else:
                         llm_label[k] = str(v)
                 successfully_labeled = True