fix: use asyncio.wait_for for backward compatibility

Author: zyzniewski

tests/e2e_tests/test_incentive.py

@@ -84,8 +84,7 @@ async def test_incentive(local_chain, subtensor, templates, alice_wallet, bob_wa
     async with templates.miner(bob_wallet, netuid):
         async with templates.validator(alice_wallet, netuid) as validator:
             # wait for the Validator to process and set_weights
-            async with asyncio.timeout(60):
-                await validator.set_weights.wait()
+            await asyncio.wait_for(validator.set_weights.wait(), 60)
 
             # Wait till new epoch
             await wait_interval(tempo, subtensor, netuid)

tests/e2e_tests/utils/e2e_test_utils.py

@@ -116,8 +116,7 @@ async def __aenter__(self):
 
             self.__reader_task = asyncio.create_task(self._reader())
 
-            async with asyncio.timeout(30):
-                await self.started.wait()
+            await asyncio.wait_for(self.started.wait(), 30)
 
             return self
 
@@ -166,8 +165,7 @@ async def __aenter__(self):
 
             self.__reader_task = asyncio.create_task(self._reader())
 
-            async with asyncio.timeout(30):
-                await self.started.wait()
+            await asyncio.wait_for(self.started.wait(), 30)
 
             return self