Merge pull request #3124 from opentensor/fix/roman/py3.14-tests-issue

[v10] Fix py3.14 tests issue

Author: basfroman

Makefile

@@ -34,4 +34,5 @@ check: ruff
 	@mypy --ignore-missing-imports bittensor/ --python-version=3.11
 	@mypy --ignore-missing-imports bittensor/ --python-version=3.12
 	@mypy --ignore-missing-imports bittensor/ --python-version=3.13
+	@mypy --ignore-missing-imports bittensor/ --python-version=3.14
 	@flake8 bittensor/ --count

tests/e2e_tests/test_axon.py

@@ -1,4 +1,5 @@
 import asyncio
+import time
 
 import pytest
 
@@ -10,8 +11,7 @@
 )
 
 
-@pytest.mark.asyncio
-async def test_axon(subtensor, templates, alice_wallet):
+def test_axon(subtensor, templates, alice_wallet):
     """
     Test the Axon mechanism and successful registration on the network with sync Subtensor.
 
@@ -45,9 +45,9 @@ async def test_axon(subtensor, templates, alice_wallet):
     assert old_axon.port == 0, f"Expected port 0, but got {old_axon.port}"
     assert old_axon.ip_type == 0, f"Expected IP type 0, but got {old_axon.ip_type}"
 
-    async with templates.miner(alice_wallet, alice_sn.netuid):
+    with templates.miner(alice_wallet, alice_sn.netuid):
         # Waiting for 5 seconds for metagraph to be updated
-        await asyncio.sleep(5)
+        time.sleep(5)
 
         # Refresh the metagraph
         metagraph = subtensor.metagraphs.metagraph(alice_sn.netuid)

tests/e2e_tests/test_dendrite.py

@@ -1,4 +1,5 @@
 import asyncio
+import time
 
 import pytest
 
@@ -21,8 +22,7 @@
 NON_FAST_RUNTIME_TEMPO = 10
 
 
-@pytest.mark.asyncio
-async def test_dendrite(subtensor, templates, alice_wallet, bob_wallet):
+def test_dendrite(subtensor, templates, alice_wallet, bob_wallet):
     """
     Test the Dendrite mechanism
 
@@ -114,8 +114,8 @@ async def test_dendrite(subtensor, templates, alice_wallet, bob_wallet):
     assert bob_neuron.validator_trust == 0.0
     assert bob_neuron.pruning_score == 0
 
-    async with templates.validator(bob_wallet, alice_sn.netuid):
-        await asyncio.sleep(5)  # wait for 5 seconds for the Validator to process
+    with templates.validator(bob_wallet, alice_sn.netuid):
+        time.sleep(5)  # wait for 5 seconds for the Validator to process
 
         subtensor.wait_for_block(
             subtensor.subnets.get_next_epoch_start_block(alice_sn.netuid) + 1

tests/e2e_tests/test_incentive.py

@@ -1,5 +1,5 @@
 import asyncio
-
+import time
 import pytest
 
 from bittensor.utils.btlogging import logging
@@ -16,8 +16,7 @@
 )
 
 
-@pytest.mark.asyncio
-async def test_incentive(subtensor, templates, alice_wallet, bob_wallet):
+def test_incentive(subtensor, templates, alice_wallet, bob_wallet):
     """
     Test the incentive mechanism and interaction of miners/validators
 
@@ -78,14 +77,11 @@ async def test_incentive(subtensor, templates, alice_wallet, bob_wallet):
     max_attempt = 3
     while True:
         try:
-            async with templates.miner(bob_wallet, alice_sn.netuid) as miner:
-                await asyncio.wait_for(miner.started.wait(), 60)
+            with templates.miner(bob_wallet, alice_sn.netuid):
+                time.sleep(5)
+                with templates.validator(alice_wallet, alice_sn.netuid):
+                    time.sleep(5)
 
-                async with templates.validator(
-                    alice_wallet, alice_sn.netuid
-                ) as validator:
-                    # wait for the Validator to process and set_weights
-                    await asyncio.wait_for(validator.set_weights.wait(), 60)
             break
         except asyncio.TimeoutError:
             if max_attempt > 0:

tests/e2e_tests/test_neuron_certificate.py

@@ -7,8 +7,7 @@
 )
 
 
-@pytest.mark.asyncio
-async def test_neuron_certificate(subtensor, alice_wallet):
+def test_neuron_certificate(subtensor, alice_wallet):
     """
     Tests the metagraph
 

tests/e2e_tests/test_root_set_weights.py

@@ -1,5 +1,5 @@
 import asyncio
-
+import time
 import pytest
 
 from bittensor.extras.dev_framework import REGISTER_NEURON
@@ -36,8 +36,7 @@
 """
 
 
-@pytest.mark.asyncio
-async def test_root_reg_hyperparams(subtensor, templates, alice_wallet, bob_wallet):
+def test_root_reg_hyperparams(subtensor, templates, alice_wallet, bob_wallet):
     """
     Test root weights and hyperparameters in the Subtensor network.
 
@@ -93,8 +92,8 @@ async def test_root_reg_hyperparams(subtensor, templates, alice_wallet, bob_wall
         period=16,
     ).success, "Unable to stake from Bob to Alice"
 
-    async with templates.validator(alice_wallet, alice_sn.netuid):
-        await asyncio.sleep(5)  # Wait a bit for chain to process data
+    with templates.validator(alice_wallet, alice_sn.netuid):
+        time.sleep(5)  # Wait a bit for chain to process data
 
         # Fetch uid against Alice's hotkey on sn 2 (it will be 0 as she is the only registered neuron)
         alice_uid_sn_2 = subtensor.subnets.get_uid_for_hotkey_on_subnet(

tests/e2e_tests/test_subtensor_functions.py

@@ -1,5 +1,7 @@
 import asyncio
 import re
+import time
+
 import pytest
 
 from bittensor.utils.balance import Balance
@@ -32,8 +34,7 @@
 """
 
 
-@pytest.mark.asyncio
-async def test_subtensor_extrinsics(subtensor, templates, alice_wallet, bob_wallet):
+def test_subtensor_extrinsics(subtensor, templates, alice_wallet, bob_wallet):
     """
     Tests subtensor extrinsics
 
@@ -165,10 +166,9 @@ async def test_subtensor_extrinsics(subtensor, templates, alice_wallet, bob_wall
         "Balance for Bob is not correct after burned register"
     )
 
-    async with templates.validator(alice_wallet, netuid):
-        await asyncio.sleep(
-            5
-        )  # wait for 5 seconds for the metagraph and subtensor to refresh with latest data
+    with templates.validator(alice_wallet, netuid):
+        # wait for 5 seconds for the metagraph and subtensor to refresh with latest data
+        time.sleep(5)
 
         alice_sn.wait_next_epoch()
 

tests/e2e_tests/utils/e2e_test_utils.py

@@ -102,16 +102,33 @@ def uninstall_templates(install_dir):
     shutil.rmtree(install_dir)
 
 
+def get_event_loop():
+    """Returns the current event loop or creates a new one if there is none."""
+    try:
+        loop = asyncio.get_running_loop()
+    except RuntimeError:
+        loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(loop)
+    return loop
+
+
 class Templates:
     class Miner:
         def __init__(self, dir, wallet, netuid):
             self.dir = dir
             self.wallet = wallet
             self.netuid = netuid
             self.process = None
-
+            self.loop = get_event_loop()
             self.started = asyncio.Event()
 
+        def __enter__(self):
+            self.loop.run_until_complete(self.__aenter__())
+            return self
+
+        def __exit__(self, exc_type, exc_value, traceback):
+            self.loop.run_until_complete(self.__aexit__(exc_type, exc_value, traceback))
+
         async def __aenter__(self):
             env = os.environ.copy()
             env["BT_LOGGING_DEBUG"] = "1"
@@ -172,6 +189,7 @@ def __init__(self, dir, wallet, netuid):
             self.netuid = netuid
             self.process = None
 
+            self.loop = get_event_loop()
             self.started = asyncio.Event()
             self.set_weights = asyncio.Event()
 
@@ -209,12 +227,19 @@ async def __aenter__(self):
 
             return self
 
+        def __enter__(self):
+            self.loop.run_until_complete(self.__aenter__())
+            return self
+
         async def __aexit__(self, exc_type, exc_value, traceback):
             self.process.terminate()
             self.__reader_task.cancel()
 
             await self.process.wait()
 
+        def __exit__(self, exc_type, exc_value, traceback):
+            self.loop.run_until_complete(self.__aexit__(exc_type, exc_value, traceback))
+
         async def _reader(self):
             async for line in self.process.stdout:
                 try: