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..d09ba27 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.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: