From 1143a3493cdc092706278c192e3b4a222d077d67 Mon Sep 17 00:00:00 2001 From: ifrit98 Date: Mon, 10 Jul 2023 22:46:25 +0000 Subject: [PATCH 1/2] pull num_uids from subtensor --- openvalidators/gating.py | 3 +-- openvalidators/neuron.py | 2 ++ 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/openvalidators/gating.py b/openvalidators/gating.py index 482ad0e..54fe341 100644 --- a/openvalidators/gating.py +++ b/openvalidators/gating.py @@ -52,8 +52,7 @@ def add_args(cls, parser: argparse.ArgumentParser): parser.add_argument( "--gating.num_uids", type=int, - default=1024, - help="Number of uids to gate on", + help="Number of uids to gate on. Default is pulled from subtensor directly", ) parser.add_argument( "--gating.learning_rate", diff --git a/openvalidators/neuron.py b/openvalidators/neuron.py index 227d2c8..5e7b095 100644 --- a/openvalidators/neuron.py +++ b/openvalidators/neuron.py @@ -108,6 +108,8 @@ def __init__(self): # Init the gating model which learns which miners to select for each query. bt.logging.debug("loading", "gating_model") + if not self.config.gating.num_uids: + self.config.gating.num_uids = self.subtensor.query_subtensor("MaxAllowedUids", params=[self.config.netuid]) if self.config.neuron.mock_gating_model: self.gating_model = MockGatingModel(self.metagraph.n.item()) elif self.config.neuron.use_custom_gating_model: From 7858c1257d2c0c9d2a17e24589047c743fc9856a Mon Sep 17 00:00:00 2001 From: ifrit98 Date: Tue, 11 Jul 2023 15:23:31 +0000 Subject: [PATCH 2/2] change maxuids query -> current uids to avoid consensus issues --- openvalidators/neuron.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/openvalidators/neuron.py b/openvalidators/neuron.py index 5e7b095..d09ba27 100644 --- a/openvalidators/neuron.py +++ b/openvalidators/neuron.py @@ -109,7 +109,7 @@ def __init__(self): # Init the gating model which learns which miners to select for each query. bt.logging.debug("loading", "gating_model") if not self.config.gating.num_uids: - self.config.gating.num_uids = self.subtensor.query_subtensor("MaxAllowedUids", params=[self.config.netuid]) + self.config.gating.num_uids = self.subtensor.subnetwork_n(self.config.netuid) if self.config.neuron.mock_gating_model: self.gating_model = MockGatingModel(self.metagraph.n.item()) elif self.config.neuron.use_custom_gating_model: