From 27221b19eb03dca40f3183d06382fd295047c3e2 Mon Sep 17 00:00:00 2001
From: isabella618033 <isabella618033@gmail.com>
Date: Wed, 16 Aug 2023 15:06:27 +0000
Subject: [PATCH] log only neuron and reward for config

---
 openvalidators/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/openvalidators/utils.py b/openvalidators/utils.py
index d008f55..e9e6c92 100644
--- a/openvalidators/utils.py
+++ b/openvalidators/utils.py
@@ -54,7 +54,7 @@ def init_wandb(self, reinit=False):
         reinit=reinit,
         project=self.config.wandb.project_name,
         entity=self.config.wandb.entity,
-        config=self.config,
+        config={key: self.config.get(key, None) for key in ('neuron', 'reward')},
         mode="offline" if self.config.wandb.offline else "online",
         dir=self.config.neuron.full_path,
         tags=tags,