Merge pull request #2879 from opentensor/fix/roman/serve_axon

Using `hotkey` instead of `coldkey` to sign extrinsic in `serve_axon`

Author: basfroman

bittensor/core/extrinsics/asyncex/serving.py

@@ -61,6 +61,7 @@ async def do_serve_axon(
         wallet=wallet,
         wait_for_inclusion=wait_for_inclusion,
         wait_for_finalization=wait_for_finalization,
+        sign_with="hotkey",
         period=period,
     )
     return success, message
@@ -140,7 +141,7 @@ async def serve_extrinsic(
         f"Serving axon with: [blue]AxonInfo({wallet.hotkey.ss58_address}, {ip}:{port})[/blue] -> "
         f"[green]{subtensor.network}:{netuid}[/green]"
     )
-    success, message = do_serve_axon(
+    success, message = await do_serve_axon(
         subtensor=subtensor,
         wallet=wallet,
         call_params=params,

bittensor/core/extrinsics/serving.py

@@ -60,6 +60,7 @@ def do_serve_axon(
         wallet=wallet,
         wait_for_inclusion=wait_for_inclusion,
         wait_for_finalization=wait_for_finalization,
+        sign_with="hotkey",
         period=period,
     )
     return success, message

tests/unit_tests/test_subtensor.py

@@ -1492,6 +1492,7 @@ def test_do_serve_axon_is_success(
         wallet=fake_wallet,
         wait_for_inclusion=fake_wait_for_inclusion,
         wait_for_finalization=fake_wait_for_finalization,
+        sign_with="hotkey",
         period=None,
     )
 
@@ -1530,6 +1531,7 @@ def test_do_serve_axon_is_not_success(subtensor, fake_wallet, mocker, fake_call_
         wallet=fake_wallet,
         wait_for_inclusion=fake_wait_for_inclusion,
         wait_for_finalization=fake_wait_for_finalization,
+        sign_with="hotkey",
         period=None,
     )
 
@@ -1568,6 +1570,7 @@ def test_do_serve_axon_no_waits(subtensor, fake_wallet, mocker, fake_call_params
         wallet=fake_wallet,
         wait_for_inclusion=fake_wait_for_inclusion,
         wait_for_finalization=fake_wait_for_finalization,
+        sign_with="hotkey",
         period=None,
     )
     assert result == (True, "")