fixes test

opentensor · Aug 6, 2024 · 8e2fe29 · 8e2fe29
1 parent 38e55a5
commit 8e2fe29
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/tests/e2e_tests/subcommands/subnet/test_list.py b/tests/e2e_tests/subcommands/subnet/test_list.py
@@ -17,14 +17,13 @@ def test_list_command(local_chain, capsys):
     # Register root as Alice
     keypair, exec_command, wallet = setup_wallet("//Alice")
 
-    netuids = [0, 3]
+    netuid = 0
 
-    assert local_chain.query("SubtensorModule", "NetworksAdded", netuids).serialize()
+    assert local_chain.query("SubtensorModule", "NetworksAdded", [netuid]).serialize()
 
     exec_command(RegisterSubnetworkCommand, ["s", "create"])
 
-    netuids.append(1)
-    netuids.sort()
+    netuid - 1
 
     # Verify subnet 1 created successfully
-    assert local_chain.query("SubtensorModule", "NetworksAdded", netuids).serialize()
+    assert local_chain.query("SubtensorModule", "NetworksAdded", [netuid]).serialize()