fix: wait_epoch didn't take tempo +1 into account when called with times arg

Author: zyzniewski-reef

tests/e2e_tests/utils/chain_interactions.py

@@ -74,7 +74,7 @@ def sudo_set_hyperparameter_values(
     return response.is_success
 
 
-async def wait_epoch(subtensor: "Subtensor", netuid: int = 1, times: int = 1):
+async def wait_epoch(subtensor: "Subtensor", netuid: int = 1, **kwargs):
     """
     Waits for the next epoch to start on a specific subnet.
 
@@ -90,7 +90,7 @@ async def wait_epoch(subtensor: "Subtensor", netuid: int = 1, times: int = 1):
         raise Exception("could not determine tempo")
     tempo = q_tempo[0].value
     logging.info(f"tempo = {tempo}")
-    await wait_interval(tempo * times, subtensor, netuid)
+    await wait_interval(tempo, subtensor, netuid, **kwargs)
 
 
 def next_tempo(current_block: int, tempo: int, netuid: int) -> int:
@@ -117,6 +117,7 @@ async def wait_interval(
     netuid: int = 1,
     reporting_interval: int = 1,
     sleep: float = 0.25,
+    times: int = 1,
 ):
     """
     Waits until the next tempo interval starts for a specific subnet.
@@ -126,7 +127,11 @@ async def wait_interval(
     the current block number until the next tempo interval starts.
     """
     current_block = subtensor.get_current_block()
-    next_tempo_block_start = next_tempo(current_block, tempo, netuid)
+    next_tempo_block_start = current_block
+
+    for _ in range(times):
+        next_tempo_block_start = next_tempo(next_tempo_block_start, tempo, netuid)
+
     last_reported = None
 
     while current_block < next_tempo_block_start: