List coldkey

src/__init__.py

@@ -6,4 +6,4 @@ TIMEZONE = ZoneInfo("America/Los_Angeles")
 START_DATE = datetime(2024, 11, 9)
 
 Uid: TypeAlias = int
-Hotkey: TypeAlias = str
+Key: TypeAlias = str

src/chain_data.py

@@ -12,7 +12,7 @@ from fiber.chain.models import Node
 from substrateinterface.storage import StorageKey
 
 from network_commitments import Decoder
-from src import Hotkey, Uid, TIMEZONE
+from src import Key, Uid, TIMEZONE
 
 DISABLE_COMMITMENTS_FETCH = int(os.getenv("DISABLE_COMMITMENTS_FETCH") or 0) > 0
 
@@ -54,13 +54,13 @@ class Commitment:
 
 SPEC_VERSION = 7
 NET_UID = 39
-WEIGHTS_BY_MINER: dict[Hotkey, list[tuple[Hotkey, Weight]]] = {}
-VALIDATOR_IDENTITIES: dict[Hotkey, str] = {}
-COMMITMENTS: dict[Hotkey, Commitment] = {}
-UPDATED: dict[Hotkey, int] = {}
+WEIGHTS_BY_MINER: dict[Key, list[tuple[Key, Weight]]] = {}
+VALIDATOR_IDENTITIES: dict[Key, str] = {}
+COMMITMENTS: dict[Key, Commitment] = {}
+UPDATED: dict[Key, int] = {}
 
-UIDS_BY_HOTKEY: dict[Hotkey, Uid] = {}
-HOTKEYS_BY_UID: dict[Uid, Hotkey] = {}
+UIDS_BY_HOTKEY: dict[Key, Uid] = {}
+HOTKEYS_BY_UID: dict[Uid, Key] = {}
 
 substrate = get_substrate()
 metagraph = Metagraph(substrate, netuid=NET_UID, load_old_nodes=False)
@@ -82,7 +82,7 @@ def is_validator(node: Node) -> bool:
     return node.vtrust > 0 or node.stake > 10_000
 
 
-def get_nodes() -> dict[Hotkey, Node]:
+def get_nodes() -> dict[Key, Node]:
     return metagraph.nodes
 
 

src/submissions.py

@@ -1,12 +1,12 @@
 import gradio as gr
 import pandas as pd
 
-from chain_data import sync_metagraph, COMMITMENTS, UIDS_BY_HOTKEY
-from src import Hotkey
+from chain_data import sync_metagraph, COMMITMENTS, UIDS_BY_HOTKEY, metagraph
+from src import Key
 from wandb_data import get_current_runs, Run
 
 
-def get_status(run: Run, hotkey: Hotkey, block: int) -> tuple[str, str]:
+def get_status(run: Run, hotkey: Key, block: int) -> tuple[str, str]:
     if hotkey in run.submissions and block > run.submissions[hotkey].info.block and hotkey not in run.invalid_submissions:
         return "Pending", "orange"
 
@@ -25,12 +25,14 @@ def create_submissions() -> gr.Dataframe:
     runs = sorted(get_current_runs(), key=lambda run: run.uid)
 
     for hotkey, commitment in COMMITMENTS.items():
+        coldkey = metagraph.nodes[hotkey].coldkey
         row = [
             UIDS_BY_HOTKEY[hotkey],
             f"[{'/'.join(commitment.get_repo_link().split('/')[-2:])}]({commitment.get_repo_link()})",
             f"[{commitment.block}](https://taostats.io/block/{commitment.block})",
             f"[{commitment.revision}]({commitment.get_repo_link()}/commit/{commitment.revision})",
             f"[{hotkey[:6]}...](https://taostats.io/hotkey/{hotkey})",
+            f"[{coldkey[:6]}...](https://taostats.io/hotkey/{coldkey})",
             commitment.contest.name,
         ]
 
@@ -42,8 +44,8 @@ def create_submissions() -> gr.Dataframe:
 
     data.sort(key=lambda x: int(x[2].split('[')[1].split(']')[0]), reverse=True)
 
-    columns = ["UID", "Model", "Block", "Revision", "Hotkey", "Contest"]
-    datatype = ["number", "markdown", "markdown", "markdown", "markdown", "markdown"]
+    columns = ["UID", "Model", "Block", "Revision", "Hotkey", "Coldkey", "Contest"]
+    datatype = ["number", "markdown", "markdown", "markdown", "markdown", "markdown", "markdown"]
     for run in runs:
         columns.append(f"{run.uid}")
         datatype.append("markdown")

src/validator_weights.py

@@ -2,7 +2,7 @@ import gradio as gr
 import pandas as pd
 
 from chain_data import WEIGHTS_BY_MINER, UIDS_BY_HOTKEY, get_nodes, sync_metagraph, Weight
-from wandb_data import Hotkey, get_current_runs
+from wandb_data import Key, get_current_runs
 
 
 def get_color_by_weight(weight: float) -> str:
@@ -23,12 +23,12 @@ def get_color_by_weight(weight: float) -> str:
         return f"rgb(0, {g}, 0)"
 
 
-def get_active_weights() -> dict[Hotkey, list[tuple[Hotkey, Weight]]]:
+def get_active_weights() -> dict[Key, list[tuple[Key, Weight]]]:
     runs = get_current_runs()
-    weights: dict[Hotkey, list[tuple[Hotkey, Weight]]] = {}
+    weights: dict[Key, list[tuple[Key, Weight]]] = {}
 
     for hotkey, validator_weights in WEIGHTS_BY_MINER.items():
-        new_weights: list[tuple[Hotkey, Weight]] = []
+        new_weights: list[tuple[Key, Weight]] = []
         for validator_hotkey, weight in validator_weights:
             if validator_hotkey in [run.hotkey for run in runs]:
                 new_weights.append((validator_hotkey, weight))

src/wandb_data.py

@@ -7,7 +7,7 @@ import wandb
 import wandb.apis.public as wapi
 from substrateinterface import Keypair
 
-from src import TIMEZONE, Hotkey
+from src import TIMEZONE, Key
 from chain_data import UIDS_BY_HOTKEY, VALIDATOR_IDENTITIES, sync_metagraph
 
 WANDB_RUN_PATH = os.environ["WANDB_RUN_PATH"]
@@ -76,8 +76,8 @@ class Run:
     winner_uid: int | None
     baseline_metrics: MetricData | None
     total_submissions: int
-    submissions: dict[Hotkey, Submission]
-    invalid_submissions: set[Hotkey]
+    submissions: dict[Key, Submission]
+    invalid_submissions: set[Key]
 
 
 RUNS: dict[str, list[Run]] = {}
@@ -127,9 +127,9 @@ def _add_runs(wandb_runs: list[wapi.Run]):
 
         metrics = wandb_run.summary
 
-        submission_info: dict[Hotkey, SubmissionInfo] = {}
-        submissions: dict[Hotkey, Submission] = {}
-        invalid_submissions: set[Hotkey] = set()
+        submission_info: dict[Key, SubmissionInfo] = {}
+        submissions: dict[Key, Submission] = {}
+        invalid_submissions: set[Key] = set()
 
         baseline_metrics: MetricData | None = None
         if "baseline" in metrics:
@@ -182,11 +182,11 @@ def _add_runs(wandb_runs: list[wapi.Run]):
                 ...
 
         status = _status_from_run(wandb_run)
-        winners = sorted(
+        winner_uid = min(
             submissions.values(),
             key=lambda submission: (submission.rank, submission.info.block),
+            default=None
         )
-        winner_uid = winners[0].info.uid if winners else None
 
         uid = int(wandb_run.config["uid"])
         hotkey = wandb_run.config["hotkey"]