Skip to content

Commit

Permalink
fix config naming in BaseNeuron
Browse files Browse the repository at this point in the history
  • Loading branch information
mccrindlebrian committed Dec 18, 2024
1 parent 1d8336e commit 39fba2d
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions atom/base/neuron.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,8 @@
import bittensor as bt

from atom.base.ttl import ttl_get_block
from atom.base.config import check_config, add_args, config
from atom.base.config import check_config, add_args
from atom.base.config import config as bittensor_config

from atom.mock.mock import MockSubtensor, MockMetagraph, create_wallet

Expand Down Expand Up @@ -48,14 +49,14 @@ def add_args(cls, parser):
add_args(cls, parser)

@classmethod
def config(cls):
def get_config(cls):
"""
Returns the default configuration for the neuron.
Returns:
bt.Config: Default configuration object
"""
return config(cls)
return bittensor_config(cls)

subtensor: "bt.subtensor"
wallet: "bt.wallet"
Expand Down Expand Up @@ -83,8 +84,8 @@ def block(self):
return ttl_get_block(self)

def __init__(self, config=None):
base_config = copy.deepcopy(config or BaseNeuron.config())
self.config = self.config()
base_config = copy.deepcopy(config or BaseNeuron.get_config())
self.config = self.get_config()
self.config.merge(base_config)
self.check_config(self.config)

Expand Down

0 comments on commit 39fba2d

Please sign in to comment.