fix

Author: Roman

tests/e2e_tests/test_commit_reveal_v3.py

@@ -33,16 +33,17 @@ async def test_commit_and_reveal_weights_cr3(local_chain, subtensor, alice_walle
     BLOCK_TIME = (
         0.25 if subtensor.is_fast_blocks() else 12.0
     )  # 12 for non-fast-block, 0.25 for fast block
-    netuid = 2
+    netuid = subtensor.get_total_subnets()  # 2
+
     logging.console.info("Testing test_commit_and_reveal_weights")
 
     # Register root as Alice
     assert subtensor.register_subnet(alice_wallet), "Unable to register the subnet"
 
     # Verify subnet 2 created successfully
-    assert subtensor.subnet_exists(netuid), "Subnet wasn't created successfully"
+    assert subtensor.subnet_exists(netuid), f"Subnet {netuid} wasn't created successfully"
 
-    logging.console.info("Subnet 2 is registered")
+    logging.console.success(f"Subnet {netuid} is registered")
 
     # Enable commit_reveal on the subnet
     assert sudo_set_hyperparameter_bool(
@@ -105,7 +106,7 @@ async def test_commit_and_reveal_weights_cr3(local_chain, subtensor, alice_walle
     )
 
     # Wait for 2 tempos to pass as CR3 only reveals weights after 2 tempos + 1
-    subtensor.wait_for_block(subtensor.block + (tempo_set * 2) + 1)
+    subtensor.wait_for_block(tempo_set * 2 + 1)
 
     # Lower than this might mean weights will get revealed before we can check them
     if upcoming_tempo - current_block < 3:

tests/e2e_tests/test_reveal_commitments.py

@@ -81,7 +81,8 @@ async def test_set_reveal_commitment(local_chain, subtensor, alice_wallet, bob_w
     # Sometimes the chain doesn't update the repository right away and the commit doesn't appear in the expected
     # `last_drand_round`. In this case need to wait a bit.
     print(f"Waiting for reveal round {target_reveal_round}")
-    while subtensor.last_drand_round() <= target_reveal_round + 1:
+    chain_offset = 1 if subtensor.is_fast_blocks() else 24
+    while subtensor.last_drand_round() <= target_reveal_round + chain_offset:
         # wait one drand period (3 sec)
         print(f"Current last reveled drand round {subtensor.last_drand_round()}")
         time.sleep(3)