wandb_logs_fix (my bad)

user-friendly-metrics.py

@@ -111,6 +111,7 @@ class UserFriendlyMetrics(evaluate.Metric):
         self,
         results,
         wandb_section: str = None,
+        wandb_runs = None
         wandb_project="user_friendly_metrics",
         log_plots: bool = True,
         debug: bool = False,
@@ -129,7 +130,20 @@ class UserFriendlyMetrics(evaluate.Metric):
         current_datetime = datetime.datetime.now()
         formatted_datetime = current_datetime.strftime("%Y-%m-%d_%H-%M-%S")
         wandb.login(key=os.getenv("WANDB_API_KEY"))
-
+        
+        if not wandb_runs is None:
+            assert len(wandb_runs) == len(results), "runs and results must have the same length"
+        else:
+            wandb_runs = [f"{i}_{formatted_datetime}" for i in list(results.keys())]
+
+        for wandb_run_name, result in zip(wandb_runs, results.values()):        
+            self.wandb_run(result = result, 
+                    wandb_run_name = wandb_run_name, 
+                    wandb_project = wandb_project, 
+                    debug = debug)
+
+    def wandb_run(self, result, wandb_run_name, wandb_project, debug):
+            
         run = wandb.init(
             project=wandb_project,
             name=f"evaluation-{formatted_datetime}",