From 6ec5e251c38200417124faee24fd02dfa7660488 Mon Sep 17 00:00:00 2001 From: isabella618033 Date: Wed, 13 Sep 2023 19:53:28 +0000 Subject: [PATCH] fixes --- neurons/miners/template/miner.py | 2 +- prompting/baseminer/blacklist.py | 17 ++++++++--------- 2 files changed, 9 insertions(+), 10 deletions(-) diff --git a/neurons/miners/template/miner.py b/neurons/miners/template/miner.py index 5efaa65..ce31628 100644 --- a/neurons/miners/template/miner.py +++ b/neurons/miners/template/miner.py @@ -45,7 +45,7 @@ def config(self) -> "bt.Config": Returns: bt.Config: A configuration object with the miner's operational parameters. """ - parser = argparse.ArgumentParser(description="OpenAI Miner Configs") + parser = argparse.ArgumentParser(description="Template Miner Configs") self.add_args(parser) return bt.config(parser) diff --git a/prompting/baseminer/blacklist.py b/prompting/baseminer/blacklist.py index 6ead4d7..899d0a0 100644 --- a/prompting/baseminer/blacklist.py +++ b/prompting/baseminer/blacklist.py @@ -67,16 +67,15 @@ def default_blacklist(self, synapse: Prompting) -> Union[Tuple[bool, str], bool] and synapse.dendrite.hotkey not in self.metagraph.hotkeys ): return True, "hotkey not registered" - - # If the user is registered, it has a UID. - uid = self.metagraph.hotkeys.index(synapse.dendrite.hotkey) - + # Check if the key has validator permit - if ( - self.config.miner.blacklist.force_validator_permit - and not self.metagraph.validator_permit[uid] - ): - return True, "validator permit required" + if self.config.miner.blacklist.force_validator_permit: + if synapse.dendrite.hotkey in self.metagraph.hotkeys: + uid = self.metagraph.hotkeys.index(synapse.dendrite.hotkey) + if not self.metagraph.validator_permit[uid]: + return True, "validator permit required" + else: + return True, "validator permit required, but hotkey not registered" if is_prompt_in_cache(self, synapse): return True, "prompt already sent recently"