diff --git a/commons/data_manager.py b/commons/data_manager.py index 23e27090..cf28a73d 100644 --- a/commons/data_manager.py +++ b/commons/data_manager.py @@ -152,7 +152,6 @@ async def validator_save(cls, scores): torch.save( { "scores": scores, - # "hotkey_to_accuracy": nonzero_hotkey_to_accuracy, }, config.neuron.full_path + "/validator_state.pt", ) @@ -166,7 +165,7 @@ async def validator_load(cls): try: # Load the state of the validator from file. state = torch.load(config.neuron.full_path + "/validator_state.pt") - return True, state["scores"], state["hotkey_to_accuracy"] + return True, state["scores"] except FileNotFoundError: bt.logging.error("Validator state file not found.") return False, None, None