Show blacklisted submissions

src/submissions.py

@@ -3,10 +3,13 @@ import pandas as pd
 
 from chain_data import sync_metagraph, COMMITMENTS, UIDS_BY_HOTKEY, metagraph
 from src import Key
-from wandb_data import get_current_runs, Run
+from wandb_data import get_current_runs, Run, BLACKLISTED_HOTKEYS, BLACKLISTED_COLDKEYS
 
 
-def get_status(run: Run, hotkey: Key, block: int) -> tuple[str, str]:
+def get_status(run: Run, hotkey: Key, coldkey: Key, block: int) -> tuple[str, str]:
+    if hotkey in BLACKLISTED_HOTKEYS or coldkey in BLACKLISTED_COLDKEYS:
+        return "Blacklisted", "gray"
+
     if hotkey in run.submissions and block > run.submissions[hotkey].info.block and hotkey not in run.invalid_submissions:
         return "Pending", "orange"
 
@@ -37,7 +40,7 @@ def create_submissions() -> gr.Dataframe:
         ]
 
         for run in runs:
-            status, color = get_status(run, hotkey, commitment.block)
+            status, color = get_status(run, hotkey, coldkey, commitment.block)
             row.append(f"<span style='color: {color}'>{status}</span>")
 
         data.append(row)

src/wandb_data.py

@@ -3,17 +3,22 @@ from dataclasses import dataclass
 from datetime import datetime, timedelta, timezone
 from enum import Enum
 
+import requests
 import wandb
 import wandb.apis.public as wapi
 from substrateinterface import Keypair
 
-from src import TIMEZONE, Key
 from chain_data import UIDS_BY_HOTKEY, VALIDATOR_IDENTITIES, sync_metagraph
+from src import TIMEZONE, Key
 
 WANDB_RUN_PATH = os.environ["WANDB_RUN_PATH"]
 
 START_DATE = datetime(2024, 11, 29)
 
+BLACKLIST_ENDPOINT = "https://edge-inputs.api.wombo.ai/blacklist"
+BLACKLISTED_COLDKEYS = set()
+BLACKLISTED_HOTKEYS = set()
+
 
 class BenchmarkStatus(Enum):
     NOT_STARTED = ("Not Started", "orange", False)
@@ -258,6 +263,18 @@ def _fetch_current_runs(wandb_api: wandb.Api):
     _add_runs(wandb_runs)
 
 
+def _fetch_blacklisted_keys():
+    response = requests.get(BLACKLIST_ENDPOINT)
+    response.raise_for_status()
+    data = response.json()
+    BLACKLISTED_COLDKEYS.clear()
+    BLACKLISTED_HOTKEYS.clear()
+    for hotkey in data["hotkeys"]:
+        BLACKLISTED_HOTKEYS.add(hotkey)
+    for coldkey in data["coldkeys"]:
+        BLACKLISTED_COLDKEYS.add(coldkey)
+
+
 last_sync: datetime = datetime.fromtimestamp(0, TIMEZONE)
 
 
@@ -274,6 +291,7 @@ def sync():
         _fetch_history(wandb_api)
     else:
         _fetch_current_runs(wandb_api)
+    _fetch_blacklisted_keys()
 
 
 def get_current_runs() -> list[Run]: