Skip to content

Commit 131d49a

Browse files
author
Roman
committed
ruff
1 parent b3e256d commit 131d49a

File tree

3 files changed

+24
-15
lines changed

3 files changed

+24
-15
lines changed

bittensor/core/async_subtensor.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1542,9 +1542,7 @@ async def get_neuron_for_pubkey_and_subnet(
15421542

15431543
params = [netuid, uid]
15441544
json_body = await self.substrate.rpc_request(
1545-
method="neuronInfo_getNeuron",
1546-
params=params,
1547-
reuse_block_hash=reuse_block
1545+
method="neuronInfo_getNeuron", params=params, reuse_block_hash=reuse_block
15481546
)
15491547

15501548
if not (result := json_body.get("result", None)):

tests/unit_tests/test_async_subtensor.py

Lines changed: 22 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1113,7 +1113,9 @@ async def test_get_neuron_for_pubkey_and_subnet_success(subtensor, mocker):
11131113
)
11141114
subtensor.substrate.rpc_request.assert_awaited_once()
11151115
subtensor.substrate.rpc_request.assert_called_once_with(
1116-
method="neuronInfo_getNeuron", params=[fake_netuid, fake_uid], reuse_block_hash=False
1116+
method="neuronInfo_getNeuron",
1117+
params=[fake_netuid, fake_uid],
1118+
reuse_block_hash=False,
11171119
)
11181120
mocked_neuron_info.assert_called_once_with(fake_result)
11191121
assert result == "fake_neuron_info"
@@ -1188,7 +1190,9 @@ async def test_get_neuron_for_pubkey_and_subnet_rpc_result_empty(subtensor, mock
11881190
reuse_block_hash=False,
11891191
)
11901192
subtensor.substrate.rpc_request.assert_called_once_with(
1191-
method="neuronInfo_getNeuron", params=[fake_netuid, fake_uid], reuse_block_hash=False
1193+
method="neuronInfo_getNeuron",
1194+
params=[fake_netuid, fake_uid],
1195+
reuse_block_hash=False,
11921196
)
11931197
mocked_get_null_neuron.assert_called_once()
11941198
assert result == "null_neuron"
@@ -1300,7 +1304,9 @@ async def test_get_delegated_no_block_hash_no_reuse(subtensor, mocker):
13001304
# Asserts
13011305
mocked_ss58_to_vec_u8.assert_called_once_with(fake_coldkey_ss58)
13021306
mocked_rpc_request.assert_called_once_with(
1303-
method="delegateInfo_getDelegated", params=[b"encoded_coldkey"], reuse_block_hash=False
1307+
method="delegateInfo_getDelegated",
1308+
params=[b"encoded_coldkey"],
1309+
reuse_block_hash=False,
13041310
)
13051311
mocked_delegated_list_from_vec_u8.assert_called_once_with(b"mocked_result")
13061312
assert result == mocked_delegated_list_from_vec_u8.return_value
@@ -1332,7 +1338,9 @@ async def test_get_delegated_with_block_hash(subtensor, mocker):
13321338
# Asserts
13331339
mocked_ss58_to_vec_u8.assert_called_once_with(fake_coldkey_ss58)
13341340
mocked_rpc_request.assert_called_once_with(
1335-
method="delegateInfo_getDelegated", params=[fake_block_hash, b"encoded_coldkey"], reuse_block_hash=False
1341+
method="delegateInfo_getDelegated",
1342+
params=[fake_block_hash, b"encoded_coldkey"],
1343+
reuse_block_hash=False,
13361344
)
13371345
mocked_delegated_list_from_vec_u8.assert_called_once_with(b"mocked_result")
13381346
assert result == mocked_delegated_list_from_vec_u8.return_value
@@ -1391,7 +1399,9 @@ async def test_get_delegated_with_empty_result(subtensor, mocker):
13911399
# Asserts
13921400
mocked_ss58_to_vec_u8.assert_called_once_with(fake_coldkey_ss58)
13931401
mocked_rpc_request.assert_called_once_with(
1394-
method="delegateInfo_getDelegated", params=[b"encoded_coldkey"], reuse_block_hash=False
1402+
method="delegateInfo_getDelegated",
1403+
params=[b"encoded_coldkey"],
1404+
reuse_block_hash=False,
13951405
)
13961406
assert result == []
13971407

@@ -2398,7 +2408,7 @@ async def test_commit_reveal_enabled(subtensor, mocker):
23982408
param_name="CommitRevealWeightsEnabled",
23992409
block_hash=block_hash,
24002410
netuid=netuid,
2401-
reuse_block=False
2411+
reuse_block=False,
24022412
)
24032413
assert result is False
24042414

@@ -2414,7 +2424,9 @@ async def test_get_subnet_reveal_period_epochs(subtensor, mocker):
24142424
)
24152425

24162426
# Call
2417-
result = await subtensor.get_subnet_reveal_period_epochs(netuid, block_hash=block_hash)
2427+
result = await subtensor.get_subnet_reveal_period_epochs(
2428+
netuid, block_hash=block_hash
2429+
)
24182430

24192431
# Assertions
24202432
mocked_get_hyperparameter.assert_awaited_once_with(
@@ -2472,7 +2484,9 @@ async def test_register_success(subtensor, mocker):
24722484
fake_netuid = 1
24732485

24742486
mocked_register_extrinsic = mocker.AsyncMock()
2475-
mocker.patch.object(async_subtensor, "register_extrinsic", mocked_register_extrinsic)
2487+
mocker.patch.object(
2488+
async_subtensor, "register_extrinsic", mocked_register_extrinsic
2489+
)
24762490

24772491
# Call
24782492
result = await subtensor.register(wallet=fake_wallet, netuid=fake_netuid)

tests/unit_tests/utils/test_async_substrate_interface.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,4 @@
88
async def test_invalid_url_raises_exception():
99
"""Test that invalid URI raises an InvalidURI exception."""
1010
with pytest.raises(InvalidURI):
11-
async_substrate_interface.AsyncSubstrateInterface(
12-
"non_existent_entry_point"
13-
)
14-
11+
async_substrate_interface.AsyncSubstrateInterface("non_existent_entry_point")

0 commit comments

Comments
 (0)