major minor mapping logic

interfaces/cap_minor_media.py

@@ -25,6 +25,21 @@ domains = {
     "media": "media"
 }
 
+
+CAP_MEDIA_CODES = list(CAP_MEDIA_NUM_DICT.values())
+CAP_MIN_CODES = list(CAP_MIN_NUM_DICT.values())
+
+major_index_to_id = {i: code for i, code in enumerate(CAP_MEDIA_CODES)}-
+minor_id_to_index = {code: i for i, code in enumerate(CAP_MIN_CODES)}
+minor_index_to_id = {i: code for i, code in enumerate(CAP_MIN_CODES)}
+
+major_to_minor_map = defaultdict(list)
+for code in CAP_MIN_CODES:
+    major_id = int(str(code)[:-2])
+    major_to_minor_map[major_id].append(code)
+major_to_minor_map = dict(major_to_minor_map)
+
+
 def check_huggingface_path(checkpoint_path: str):
     try:
         hf_api = HfApi(token=HF_TOKEN)