diff --git a/openvalidators/utils.py b/openvalidators/utils.py index e9e6c92..c9d12f8 100644 --- a/openvalidators/utils.py +++ b/openvalidators/utils.py @@ -49,12 +49,15 @@ def init_wandb(self, reinit=False): if self.config.neuron.disable_log_rewards: tags.append("disable_log_rewards") + wandb_config = {key: copy.deepcopy(self.config.get(key, None)) for key in ('neuron', 'reward', 'netuid', 'wandb')} + wandb_config['neuron'].pop('full_path', None) + self.wandb = wandb.init( anonymous="allow", reinit=reinit, project=self.config.wandb.project_name, entity=self.config.wandb.entity, - config={key: self.config.get(key, None) for key in ('neuron', 'reward')}, + config=wandb_config, mode="offline" if self.config.wandb.offline else "online", dir=self.config.neuron.full_path, tags=tags,