debug

arena_elo/elo_rating/clean_battle_data.py

@@ -59,8 +59,7 @@ for i in range(len(IDENTITY_WORDS)):
     IDENTITY_WORDS[i] = IDENTITY_WORDS[i].lower()
 
 def parse_model_name(model_name):
-    model_source, *rest = model_name.split("_", 1)
-    model_type, model_name = rest[-1], "_".join(rest[:-1])
+    return NotImplementedError()
     return model_source, model_name, model_type
 
 def remove_html(raw):
@@ -233,8 +232,8 @@ def clean_battle_data(
             valid = True
             for _model in models:
                 try:
-                    #platform, model_name, task = _model.split("_")
-                    platform, model_name, task = parse_model_name(_model)
+                    platform, model_name, task = _model.split("_")
+                    #platform, model_name, task = parse_model_name(_model)
                 except ValueError:
                     valid = False
                     break
@@ -245,8 +244,8 @@ def clean_battle_data(
                 ct_invalid += 1
                 continue
             for i, _model in enumerate(models):
-                #platform, model_name, task = _model.split("_")
-                platform, model_name, task = parse_model_name(_model)
+                platform, model_name, task = _model.split("_")
+                #platform, model_name, task = parse_model_name(_model)
                 models[i] = model_name
                 
             # if not all(x.startswith("imagenhub_") and x.endswith("_edition") for x in models):
@@ -259,8 +258,8 @@ def clean_battle_data(
             valid = True
             for _model in models:
                 try:
-                    #platform, model_name, task = _model.split("_")
-                    platform, model_name, task = parse_model_name(_model)
+                    platform, model_name, task = _model.split("_")
+                    #platform, model_name, task = parse_model_name(_model)
                 except ValueError:
                     valid = False
                     break
@@ -271,8 +270,8 @@ def clean_battle_data(
                 ct_invalid += 1
                 continue
             for i, _model in enumerate(models):
-                #platform, model_name, task = _model.split("_")
-                platform, model_name, task = parse_model_name(_model)
+                platform, model_name, task = _model.split("_")
+                #platform, model_name, task = parse_model_name(_model)
                 models[i] = model_name
             # if not all("playground" in x.lower() or (x.startswith("imagenhub_") and x.endswith("_generation")) for x in models):
             #     print(f"Invalid model names: {models}")
@@ -288,8 +287,8 @@ def clean_battle_data(
             valid = True
             for _model in models:
                 try:
-                    #platform, model_name, task = _model.split("_")
-                    platform, model_name, task = parse_model_name(_model)
+                    platform, model_name, task = _model.split("_")
+                    #platform, model_name, task = parse_model_name(_model)
                 except ValueError:
                     valid = False
                     break
@@ -300,8 +299,8 @@ def clean_battle_data(
                 ct_invalid += 1
                 continue
             for i, _model in enumerate(models):
-                #platform, model_name, task = _model.split("_")
-                platform, model_name, task = parse_model_name(_model)
+                platform, model_name, task = _model.split("_")
+                #platform, model_name, task = parse_model_name(_model)
                 models[i] = model_name
 
         else:

model/models/__init__.py

@@ -33,9 +33,8 @@ def load_pipeline(model_name):
         the name is the name of the model used to load the model
         the type is the type of the model, either generation or edition
     """
-    #model_source, model_name, model_type = model_name.split("_")
-    model_source, *rest = model_name.split("_", 1)
-    model_type, model_name = rest[-1], "_".join(rest[:-1])
+    model_source, model_name, model_type = model_name.split("_")
+    # TODO ^Support model_name that contains _ in the name
     print("model_source, model_name, model_type =", model_source, model_name, model_type)
     if model_source == "imagenhub":
         pipe = load_imagenhub_model(model_name, model_type)