minor, minor-media label name split

label_dicts.py

@@ -21,7 +21,7 @@ def get_num_dict(df, task):
     num_dict = dict(zip(task_df["index"], task_df["code"]))
     return num_dict
 
-# NOTE: this is the same logic as in the Babel pipeline (different vairable names!)
+# NOTE: this is the same logic as in the Babel pipeline 
 # key: task type (as in codebooks.codebooks_full)
 # value: (tuple) name of label_names, num_dict variables
 # TO-DO: we could replace all of these variables with one dict-like object
@@ -48,3 +48,9 @@ for task, var_names in task_names.items():
     num_dict_var = var_names[1]
     globals()[label_name_var] = get_label_names(df_codebooks, task)
     globals()[num_dict_var] = get_num_dict(df_codebooks, task)
+    
+    
+# making it prettier
+CAP_MIN_LABEL_NAMES = {code:label_name.split("- ")[-1] for code, label_name in CAP_MIN_LABEL_NAMES.items()}
+CAP_MIN_MEDIA_LABEL_NAMES = {code:label_name.split("- ")[-1] for code, label_name in CAP_MIN_MEDIA_LABEL_NAMES.items()}
+