testing now

src/submission/submit.py

@@ -18,11 +18,7 @@ def add_new_eval(
     model_name: str,
     preds_path: str,
     track: str,
-    base_model: str,
     revision: str,
-    precision: str,
-    weight_type: str,
-    model_type: str,
 ):
     global REQUESTED_MODELS
     global USERS_TO_SUBMISSION_DATES
@@ -35,12 +31,8 @@ def add_new_eval(
         user_name = model_name.split("/")[0]
         model_path = model_name.split("/")[1]
 
-    precision = precision.split(" ")[0]
     current_time = datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ")
 
-    if model_type is None or model_type == "":
-        return styled_error("Please select a model type.")
-
     if preds_path is None or preds_path == "":
         return styled_error("Please enter a URL where your predictions file can be downloaded.")
     
@@ -51,31 +43,12 @@ def add_new_eval(
     if revision == "":
         revision = "main"
 
-    # Is the model on the hub?
-    if weight_type in ["Delta", "Adapter"]:
-        base_model_on_hub, error, _ = is_model_on_hub(model_name=base_model, revision=revision, token=TOKEN, test_tokenizer=True)
-        if not base_model_on_hub:
-            return styled_error(f'Base model "{base_model}" {error}')
-
-    if not weight_type == "Adapter":
-        model_on_hub, error, _ = is_model_on_hub(model_name=model_name, revision=revision, token=TOKEN, test_tokenizer=True)
-        if not model_on_hub:
-            return styled_error(f'Model "{model_name}" {error}')
-
     # Is the model info correctly filled?
     try:
         model_info = API.model_info(repo_id=model_name, revision=revision)
     except Exception:
         return styled_error("Could not get your model information. Please fill it up properly.")
 
-    model_size = get_model_size(model_info=model_info, precision=precision)
-
-    # Were the model card and license filled?
-    try:
-        license = model_info.cardData["license"]
-    except Exception:
-        return styled_error("Please select a license for your model")
-
     modelcard_OK, error_msg = check_model_card(model_name)
     if not modelcard_OK:
         return styled_error(error_msg)
@@ -87,27 +60,20 @@ def add_new_eval(
         "model_name": model_name,
         "preds_path": preds_path,
         "track": track,
-        "base_model": base_model,
         "revision": revision,
-        "precision": precision,
-        "weight_type": weight_type,
         "status": "PENDING",
         "submitted_time": current_time,
-        "model_type": model_type,
-        "likes": model_info.likes,
-        "params": model_size,
-        "license": license,
         "private": False,
     }
 
     # Check for duplicate submission
-    if f"{model_name}_{revision}_{precision}" in REQUESTED_MODELS:
+    if f"{model_name}_{revision}_{track}" in REQUESTED_MODELS:
         return styled_warning("This model has been already submitted.")
 
     print("Creating eval file")
     OUT_DIR = f"{EVAL_REQUESTS_PATH}/{user_name}"
     os.makedirs(OUT_DIR, exist_ok=True)
-    out_path = f"{OUT_DIR}/{model_path}_eval_request_False_{precision}_{weight_type}.json"
+    out_path = f"{OUT_DIR}/{model_path}_eval_request_False_{track}.json"
 
     with open(out_path, "w") as f:
         f.write(json.dumps(eval_entry))