Update wandb.py

Author: Samoed

autointent/_callbacks/wandb.py

@@ -113,7 +113,7 @@ def log_final_metrics(self, metrics: dict[str, Any]) -> None:
 
         try:
             print(metrics)
-            config = metrics["config"]
+            config = metrics["configs"]
             self.wandb.init(config=config, **wandb_run_init_args)
             self.wandb.log(metrics)
         except Exception as e: