Update the expected fields

skit-ai · Dec 27, 2023 · 6f9dba2 · 6f9dba2
1 parent 29efa8b
commit 6f9dba2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/skit_labels/constants.py b/skit_labels/constants.py
@@ -123,7 +123,7 @@
 VALUE = "value"
 
 EXPECTED_COLUMNS_MAPPING = {
-    "conversation_tagging": ['scenario', 'scenario_category', 'situation_str', 'call']
+    "conversation_tagging": ['scenario', 'scenario_category', 'situation_str', 'call', 'data_label']
 }
 
 CONVERSATION_TAGGING = 'conversation_tagging'