Include revision

src/chain_data.py

@@ -9,6 +9,7 @@ from fiber.chain.commitments import _deserialize_commitment_field
 from fiber.chain.interface import get_substrate
 from fiber.chain.metagraph import Metagraph
 from fiber.chain.models import Node
+from substrateinterface import SubstrateInterface
 from substrateinterface.storage import StorageKey
 
 from network_commitments import Decoder
@@ -63,7 +64,7 @@ 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)
+metagraph = Metagraph(substrate, netuid=str(NET_UID), load_old_nodes=False)
 
 
 def query_subtensor(storage_keys: list[StorageKey], block: int) -> list:
@@ -214,4 +215,4 @@ def sync_metagraph():
             fetch_commitments(block)
     except Exception as e:
         print(f"Error occurred while syncing metagraph: {e}")
-        substrate = get_substrate(subtensor_address=substrate.url)
+        substrate = SubstrateInterface(substrate.url)

src/submissions.py

@@ -16,12 +16,12 @@ class SubmissionStatus(Enum):
     INVALID = ("Invalid", "red")
 
     @staticmethod
-    def get_status(run: Run, hotkey: Key, coldkey: Key, block: int) -> "SubmissionStatus":
+    def get_status(run: Run, hotkey: Key, coldkey: Key, block: int, revision: str) -> "SubmissionStatus":
         if hotkey in BLACKLISTED_HOTKEYS or coldkey in BLACKLISTED_COLDKEYS:
             return SubmissionStatus.BLACKLISTED
 
         if any(
-            submission.hotkey == hotkey
+            submission.hotkey == hotkey and submission.revision == revision
             for submission in DUPLICATE_SUBMISSIONS
         ):
             return SubmissionStatus.DUPLICATE
@@ -61,7 +61,7 @@ def create_submissions(submission_filters: list[str]) -> gr.Dataframe:
 
         has_data = False
         for run in runs:
-            status = SubmissionStatus.get_status(run, hotkey, coldkey, commitment.block)
+            status = SubmissionStatus.get_status(run, hotkey, coldkey, commitment.block, commitment.revision)
             if status.value[0] in submission_filters:
                 row.append(f"<span style='color: {status.value[1]}'>{status.value[0]}</span>")
                 has_data = True

src/wandb_data.py

@@ -24,8 +24,9 @@ BLACKLISTED_HOTKEYS = set()
 
 class DuplicateSubmission(BaseModel):
     hotkey: Key
-    copy_of: Key
     url: str
+    revision: str
+    copy_of: Key
 
 DUPLICATE_SUBMISSIONS: list[DuplicateSubmission] = []