Merge pull request #137 from opentensor/hotfix/wandb-config

Brings back netuid and wandb to logged config
opentensor · Aug 28, 2023 · 79ace50 · 79ace50
2 parents 3f87902 + baf3f63
commit 79ace50
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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,