fix

backend/submission_handler.py

@@ -170,20 +170,20 @@ class SlackNotifier:
             }
         ]
         
-        if submission_data:
-            blocks.append({
-                "type": "section",
-                "fields": [
-                    {
-                        "type": "mrkdwn",
-                        "text": f"*Precision:*\n{submission_data.get('precision', 'UNK')}"
-                    },
-                    {
-                        "type": "mrkdwn",
-                        "text": f"*Revision:*\n{submission_data.get('revision', 'main')}"
-                    }
-                ]
-            })
+        # if submission_data:
+        #     blocks.append({
+        #         "type": "section",
+        #         "fields": [
+        #             {
+        #                 "type": "mrkdwn",
+        #                 "text": f"*Precision:*\n{submission_data.get('precision', 'UNK')}"
+        #             },
+        #             {
+        #                 "type": "mrkdwn",
+        #                 "text": f"*Revision:*\n{submission_data.get('revision', 'main')}"
+        #             }
+        #         ]
+        #     })
         
         blocks.append({
             "type": "context",
@@ -390,7 +390,7 @@ def integrate_with_submission(original_submit_func):
                 submission_data = {
                     "model": args[0] if len(args) > 0 else kwargs.get("model_name"),
                     # "base_model": args[1] if len(args) > 1 else kwargs.get("base_model"),
-                    "revision": args[2] if len(args) > 2 else kwargs.get("revision"),
+                    "revision": "main", # args[2] if len(args) > 2 else kwargs.get("revision"),
                     # "precision": args[3] if len(args) > 3 else kwargs.get("precision"),
                     # "weight_type": args[4] if len(args) > 4 else kwargs.get("weight_type"),
                     "submitted_time": datetime.utcnow().isoformat() + "Z",
@@ -522,7 +522,7 @@ def check_org_threshold(org_name: str) -> Tuple[bool, str]:
 @integrate_with_submission
 def submit_model(
     model_name: str,
-    base_model: Optional[str] = None,
+    # base_model: Optional[str] = None,
     revision: Optional[str] = "main",
     # precision: str = "",
     # weight_type: str = "",
@@ -534,8 +534,8 @@ def submit_model(
     """
     # --- 1. Input Sanitization ---
     model_name = model_name.strip()
-    if base_model:
-        base_model = base_model.strip()
+    # if base_model:
+    #     base_model = base_model.strip()
     revision = revision.strip() or "main"
     # precision = precision.strip()
     model_type = MODEL_TYPE_NORMALIZATION.get(model_type.strip().lower(), model_type.strip())