Move values to constants

skit-ai · Dec 27, 2023 · 29efa8b · 29efa8b
1 parent a0f1566
commit 29efa8b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/skit_labels/constants.py b/skit_labels/constants.py
@@ -124,4 +124,6 @@
 
 EXPECTED_COLUMNS_MAPPING = {
     "conversation_tagging": ['scenario', 'scenario_category', 'situation_str', 'call']
-}
+}
+
+CONVERSATION_TAGGING = 'conversation_tagging'
diff --git a/skit_labels/utils.py b/skit_labels/utils.py
@@ -137,7 +137,7 @@ def validate_headers(input_file, tagging_type):
 def validate_input_data(tagging_type, input_file):
     is_valid = True
     error = ''
-    if tagging_type == 'conversation_tagging':
+    if tagging_type == const.CONVERSATION_TAGGING:
         is_match, mismatch_headers = validate_headers(input_file, tagging_type)
         if not is_match:
             error = f'Headers in the input file does not match the expected fields. Mismatched fields = {mismatch_headers}'