From 3b849dc550e0c9b57730c371f68ec9ec19ba2d76 Mon Sep 17 00:00:00 2001
From: eclipsevortex <eclipsevortext@hotmail.com>
Date: Mon, 1 Apr 2024 16:14:24 +0100
Subject: [PATCH] fix: check validator has enough stake to set weight on the
 network

---
 requirements.txt            |  3 ++-
 template/base/validator.py  |  8 ++++++++
 template/utils/substrate.py | 14 ++++++++++++++
 3 files changed, 24 insertions(+), 1 deletion(-)
 create mode 100644 template/utils/substrate.py

diff --git a/requirements.txt b/requirements.txt
index c1b866e9..e5d84109 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,2 +1,3 @@
 bittensor
-torch
\ No newline at end of file
+torch
+substrateinterface
\ No newline at end of file
diff --git a/template/base/validator.py b/template/base/validator.py
index 79579f50..a81a8408 100644
--- a/template/base/validator.py
+++ b/template/base/validator.py
@@ -31,6 +31,7 @@
 from template.base.neuron import BaseNeuron
 from template.mock import MockDendrite
 from template.utils.config import add_validator_args
+from template.utils.substrate import get_weights_min_stake
 
 
 class BaseValidatorNeuron(BaseNeuron):
@@ -221,6 +222,13 @@ def set_weights(self):
         """
         Sets the validator weights to the metagraph hotkeys based on the scores it has received from the miners. The weights determine the trust and incentive level the validator assigns to miner nodes on the network.
         """
+        validator_stake = self.metagraph.S[self.uid]
+        weight_min_stake = get_weights_min_stake(self.subtensor.substrate)
+        if validator_stake < weight_min_stake:
+            bt.logging.warning(
+                f"Not enough stake t{validator_stake} to set weight, require a minimum of t{weight_min_stake}. Please stake more if you do not want to be de-registered!"
+            )
+            return
 
         # Check if self.scores contains any NaN values and log a warning if it does.
         if torch.isnan(self.scores).any():
diff --git a/template/utils/substrate.py b/template/utils/substrate.py
new file mode 100644
index 00000000..121e4e9b
--- /dev/null
+++ b/template/utils/substrate.py
@@ -0,0 +1,14 @@
+import bittensor as bt
+from substrateinterface import SubstrateInterface
+
+def get_weights_min_stake(substrate: SubstrateInterface):
+    """
+    Return the minimum of TAO a validator need to have the set weight
+    """
+    weight_min_stake = substrate.query(
+        module="SubtensorModule", storage_function="WeightsMinStake", params=[]
+    )
+    bt.logging.debug(f"get_weights_min_stake() {weight_min_stake}")
+
+    # Convert Rao to Tao
+    return int(float(weight_min_stake.value) * 10**-9)