wandb section

user-friendly-metrics.py

@@ -142,7 +142,7 @@ class UserFriendlyMetrics(evaluate.Metric):
                     wandb_project = wandb_project, 
                     debug = debug)
 
-    def wandb_run(self, result, wandb_run_name, wandb_project, debug):
+    def wandb_run(self, result, wandb_run_name, wandb_project, debug, wandb_section = None, log_plots = True):
             
         run = wandb.init(
             project = wandb_project,