Merge pull request #2703 from opentensor/tests/zyzniewski/fix_e2e_test_set_weights

Fix E2E test_set_weights

Author: zyzniewski-reef

tests/e2e_tests/test_set_weights.py

@@ -1,14 +1,13 @@
 import numpy as np
 import pytest
 
-import asyncio
-
 from bittensor.utils.balance import Balance
 from bittensor.utils.weight_utils import convert_weights_and_uids_for_emit
 from tests.e2e_tests.utils.chain_interactions import (
     sudo_set_hyperparameter_bool,
     sudo_set_hyperparameter_values,
     sudo_set_admin_utils,
+    wait_epoch,
 )
 
 
@@ -120,7 +119,7 @@ async def test_set_weights_uses_next_nonce(local_chain, subtensor, alice_wallet)
         assert success is True, f"Failed to set weights for subnet {netuid}"
 
     # Wait for the txs to be included in the chain
-    await asyncio.sleep(4)
+    await wait_epoch(subtensor, netuid=netuids[-1], times=4)
 
     for netuid in netuids:
         # Query the Weights storage map for all three subnets

tests/e2e_tests/utils/chain_interactions.py

@@ -67,7 +67,7 @@ def sudo_set_hyperparameter_values(
     return response.is_success
 
 
-async def wait_epoch(subtensor: "Subtensor", netuid: int = 1):
+async def wait_epoch(subtensor: "Subtensor", netuid: int = 1, times: int = 1):
     """
     Waits for the next epoch to start on a specific subnet.
 
@@ -83,7 +83,7 @@ async def wait_epoch(subtensor: "Subtensor", netuid: int = 1):
         raise Exception("could not determine tempo")
     tempo = q_tempo[0].value
     logging.info(f"tempo = {tempo}")
-    await wait_interval(tempo, subtensor, netuid)
+    await wait_interval(tempo * times, subtensor, netuid)
 
 
 def next_tempo(current_block: int, tempo: int, netuid: int) -> int: