Use chain identity

app.py

@@ -5,36 +5,30 @@ from enum import Enum
 from time import sleep
 from zoneinfo import ZoneInfo
 
+import bittensor as bt
 import gradio as gr
 from packaging import version
 from substrateinterface import Keypair
 from wandb.apis.importers import wandb
 from wandb.apis.public import Run
 
-HOTKEYS_TO_KNOWN_VALIDATORS: dict[str, str] = {
- "5GKH9FPPnWSUoeeTJp19wVtd84XqFW4pyK2ijV2GsFbhTrP1": "Taostats & Corcel",
- "5F4tQyWrhfGVcNhoqeiNsR6KjD4wMZ2kfhLj4oHYuyHbZAc3": "Openτensor Foundaτion",
- "5FFApaS75bv5pJHfAp2FVLBj9ZaXuFDjEypsaBNc1wCfe52v": "RoundTable21",
- "5HEo565WAy4Dbq3Sv271SAi7syBSofyfhhwRNjFNSM2gP9M2": "Foundry",
- "5HK5tp6t2S59DywmHRWPBVJeJ86T61KjurYqeooqj8sREpeN": "Bittensor Guru",
- "5GP7c3fFazW9GXK8Up3qgu2DJBk8inu4aK9TZy3RuoSWVCMi": "Datura",
- "5EhvL1FVkQPpMjZX4MAADcW42i3xPSF1KiCpuaxTYVr28sux": "TAO-Validator.com",
- "5E4z3h9yVhmQyCFWNbY9BPpwhx4xFiPwq3eeqmBgVF6KULde": "Tensorplex Labs",
- "5Hb63SvXBXqZ8zw6mwW1A39fHdqUrJvohXgepyhp2jgWedSB": "Miner's Union Validator",
- "5CXRfP2ekFhe62r7q3vppRajJmGhTi7vwvb2yr79jveZ282w": "Rizzo",
- "5CVS9d1NcQyWKUyadLevwGxg6LgBcF9Lik6NSnbe5q59jwhE": "Ary van der Touw",
- "5Fq5v71D4LX8Db1xsmRSy6udQThcZ8sFDqxQFwnUZ1BuqY5A": "NorthTensor",
- "5CsvRJXuR955WojnGMdok1hbhffZyB4N5ocrv82f3p5A2zVp": "Owl Ventures",
- "5HNQURvmjjYhTSksi8Wfsw676b4owGwfLR2BFAQzG7H3HhYf": "Neural Internet",
- "5DvTpiniW9s3APmHRYn8FroUWyfnLtrsid5Mtn5EwMXHN2ed": "FirstTensor.com",
-}
-
 WANDB_RUN_PATH = os.environ["WANDB_RUN_PATH"]
 
 REFRESH_RATE = 60 * 5 # 5 minutes
+NET_UID = 39
 AVERAGE_BENCHMARK_TIME_WARNING_THRESHOLD = 600 # 10 minutes
 ETA_WARNING_THRESHOLD = 43200 # 12 hours
 
+subtensor = bt.subtensor()
+metagraph = bt.metagraph(netuid=NET_UID)
+bt.logging.disable_logging()
+
+runs: dict[int, list[Run]] = {}
+validator_identities: dict[int, str] = {}
+
+wandb_api = wandb.Api()
+demo = gr.Blocks(css=".typewriter {font-family: 'JMH Typewriter', sans-serif;}", fill_height=True, fill_width=True)
+
 
 class Status(Enum):
  IN_PROGRESS = ("In Progress", "orange")
@@ -62,8 +56,6 @@ class State:
  average_benchmark_time: float
 
 
-wandb_api = wandb.Api()
-demo = gr.Blocks(css=".typewriter {font-family: 'JMH Typewriter', sans-serif;}", fill_height=True, fill_width=True)
 data: dict[int, State] = {}
 
 
@@ -144,6 +136,22 @@ def fetch_wandb_data():
  data = dict(sorted(data.items()))
 
 
+def fetch_identities():
+ validator_identities.clear()
+ for uid in range(metagraph.n.item()):
+ if metagraph.validator_trust[uid]:
+ identity = subtensor.substrate.query('SubtensorModule', 'Identities', [metagraph.coldkeys[uid]])
+ if identity != None:
+ validator_identities[uid] = identity.value["name"]
+
+
+def get_validator_name(validator_uid: int) -> str:
+ if validator_uid in validator_identities:
+ return validator_identities[validator_uid]
+ else:
+ return metagraph.hotkeys[validator_uid]
+
+
 def get_latest_version() -> str:
  latest_version = version.parse("0.0.0")
  for source_validator_uid, state in data.items():
@@ -154,7 +162,9 @@ def get_latest_version() -> str:
 
 
 def refresh():
+ metagraph.sync(subtensor=subtensor)
  fetch_wandb_data()
+ fetch_identities()
  demo.clear()
  now = datetime.now(tz=ZoneInfo("America/New_York"))
 
@@ -163,7 +173,7 @@ def refresh():
  elements: list[tuple] = []
  latest_version = get_latest_version()
 
- for source_validator_uid, state in data.items():
+ for uid, state in data.items():
  eta = int(state.average_benchmark_time * (state.submissions - (state.benchmarks + state.invalid)))
  time_left = timedelta(seconds=eta)
  eta_date = now + time_left
@@ -172,8 +182,8 @@ def refresh():
  average_time_text = f"{timedelta(seconds=int(state.average_benchmark_time))}" if state.average_benchmark_time else state.status.get_alt_time_text(),
 
  elements.append((
- source_validator_uid,
- HOTKEYS_TO_KNOWN_VALIDATORS.get(state.hotkey, state.hotkey),
+ uid,
+ get_validator_name(uid),
  f"<span style='color: {'springgreen' if state.version == latest_version else 'red'}'>{state.version}</span>",
  f"<span style='color: {state.status.color()}'>{state.status.name()}</span>",
  state.benchmarks + state.invalid,

requirements.txt

@@ -1,5 +1,6 @@
 gradio
 wandb
+bittensor
 substrate-interface
 packaging
 polars