Fix tests

thewhaleking · thewhaleking · commit 20d5d68b0a0f · 2025-07-10T22:38:57.000+02:00
diff --git a/bittensor_cli/src/bittensor/subtensor_interface.py b/bittensor_cli/src/bittensor/subtensor_interface.py
@@ -1547,7 +1547,8 @@ async def get_stake_fee(
             block_hash=block_hash,
         )
         result = Balance.from_rao(result_)
-        result.set_unit(origin_netuid)
+        if isinstance(origin_netuid, int):
+            result.set_unit(origin_netuid)
 
         return result
 
diff --git a/tests/e2e_tests/test_unstaking.py b/tests/e2e_tests/test_unstaking.py
@@ -231,7 +231,7 @@ def test_unstaking(local_chain, wallet_setup):
                 "144",
             ],
         )
-        assert "✅ Finalized" in stake_result.stdout
+        assert "✅ Finalized" in stake_result.stdout, stake_result.stderr
 
     stake_list = exec_command_bob(
         command="stake",

Original file line number	Diff line number	Diff line change
`@@ -1547,7 +1547,8 @@ async def get_stake_fee(`
`1547`	`1547`	`block_hash=block_hash,`
`1548`	`1548`	`)`
`1549`	`1549`	`result = Balance.from_rao(result_)`
`1550`		`- result.set_unit(origin_netuid)`
	`1550`	`+ if isinstance(origin_netuid, int):`
	`1551`	`+ result.set_unit(origin_netuid)`
`1551`	`1552`
`1552`	`1553`	`return result`
`1553`	`1554`
Original file line number	Diff line number	Diff line change
`@@ -231,7 +231,7 @@ def test_unstaking(local_chain, wallet_setup):`
`231`	`231`	`"144",`
`232`	`232`	`],`
`233`	`233`	`)`
`234`		`- assert "✅ Finalized" in stake_result.stdout`
	`234`	`+ assert "✅ Finalized" in stake_result.stdout, stake_result.stderr`
`235`	`235`
`236`	`236`	`stake_list = exec_command_bob(`
`237`	`237`	`command="stake",`