@@ -49,7 +49,7 @@ def test_metagraph(subtensor, alice_wallet, bob_wallet, dave_wallet):
49
49
alice_subnet_netuid = 2
50
50
51
51
# Register the subnet through Alice
52
- assert subtensor .register_subnet (alice_wallet ), "Unable to register the subnet"
52
+ assert subtensor .register_subnet (alice_wallet , True , True ), "Unable to register the subnet"
53
53
54
54
# Verify subnet was created successfully
55
55
assert subtensor .subnet_exists (alice_subnet_netuid ), (
@@ -74,9 +74,6 @@ def test_metagraph(subtensor, alice_wallet, bob_wallet, dave_wallet):
74
74
# Refresh the metagraph
75
75
metagraph .sync (subtensor = subtensor )
76
76
77
- # wait for updated information to arrive (important for low resource docker)
78
- subtensor .wait_for_block (subtensor .block + 10 )
79
-
80
77
# Assert metagraph has Alice and Bob neurons
81
78
assert len (metagraph .uids ) == 2 , "Metagraph doesn't have exactly 2 neurons"
82
79
assert metagraph .hotkeys [0 ] == alice_wallet .hotkey .ss58_address , (
@@ -97,9 +94,13 @@ def test_metagraph(subtensor, alice_wallet, bob_wallet, dave_wallet):
97
94
98
95
# Fetch neuron info of Bob through subtensor and metagraph
99
96
neuron_info_bob = subtensor .neuron_for_uid (uid , netuid = alice_subnet_netuid )
97
+ print (">>> neuron_info_bob" , neuron_info_bob )
100
98
metagraph_dict = neuron_to_dict (metagraph .neurons [uid ])
101
99
subtensor_dict = neuron_to_dict (neuron_info_bob )
102
100
101
+ print (">>> metagraph_dict" , metagraph_dict )
102
+ print (">>> subtensor_dict" , subtensor_dict )
103
+
103
104
# Verify neuron info is the same in both objects
104
105
assert metagraph_dict == subtensor_dict , (
105
106
"Neuron info of Bob doesn't match b/w metagraph & subtensor"
0 commit comments