move env vars check to main()

backend-developers-ltd · Dec 7, 2024 · 393c31d · 393c31d
1 parent 415d65c
commit 393c31d
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/app/bittensor_prometheus/read_wallet_and_substitute_config.py b/app/bittensor_prometheus/read_wallet_and_substitute_config.py
@@ -7,9 +7,6 @@
 BITTENSOR_WALLET_NAME = os.environ.get("BITTENSOR_WALLET_NAME")
 BITTENSOR_WALLET_HOTKEY_NAME = os.environ.get("BITTENSOR_WALLET_HOTKEY_NAME")
 
-if not BITTENSOR_WALLET_NAME or not BITTENSOR_WALLET_HOTKEY_NAME:
-    raise RuntimeError("You must set BITTENSOR_WALLET_NAME and BITTENSOR_WALLET_HOTKEY_NAME env vars")
-
 
 def get_wallet() -> bittensor.wallet:
     wallet = bittensor.wallet(
@@ -27,6 +24,8 @@ def read_and_substitute_config(hotkey: str):
 
 
 def main():
+    if not BITTENSOR_WALLET_NAME or not BITTENSOR_WALLET_HOTKEY_NAME:
+        raise RuntimeError("You must set BITTENSOR_WALLET_NAME and BITTENSOR_WALLET_HOTKEY_NAME env vars")
     wallet = get_wallet()
     read_and_substitute_config(wallet.hotkey.ss58_address)