Merge pull request #2493 from opentensor/test/axon-args

Enable arguments to be set in axon config
opentensor · Nov 28, 2024 · 7ca4d6d · 7ca4d6d
2 parents 2c3bb63 + 69a702a
commit 7ca4d6d
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/bittensor/core/axon.py b/bittensor/core/axon.py
@@ -332,11 +332,11 @@ def __init__(
         if config is None:
             config = Axon.config()
         config = copy.deepcopy(config)
-        config.axon.ip = ip or DEFAULTS.axon.ip
-        config.axon.port = port or DEFAULTS.axon.port
-        config.axon.external_ip = external_ip or DEFAULTS.axon.external_ip
-        config.axon.external_port = external_port or DEFAULTS.axon.external_port
-        config.axon.max_workers = max_workers or DEFAULTS.axon.max_workers
+        config.axon.ip = ip or config.axon.ip
+        config.axon.port = port or config.axon.port
+        config.axon.external_ip = external_ip or config.axon.external_ip
+        config.axon.external_port = external_port or config.axon.external_port
+        config.axon.max_workers = max_workers or config.axon.max_workers
         Axon.check_config(config)
         self.config = config  # type: ignore