Add resume wandb reporter

fl4health/reporting/wandb_reporter.py

@@ -30,6 +30,7 @@ def __init__(
         tags: list[str] | None = None,
         name: str | None = None,
         id: str | None = None,
+        resume: str = "allow",
         **kwargs: Any,
     ) -> None:
         """
@@ -56,7 +57,11 @@ def __init__(
             name (str | None, optional): A short display name for this run. Default generates a random two-word name.
             id (str | None, optional): A unique ID for this run. It must be unique in the project, and if you delete a
                 run you can't reuse the ID.
-            kwargs (Any):  Keyword arguments to wandb.init excluding the ones explicitly described above.
+            resume (str): Indicates how to handle the case when a run has the same entity, project and run id as
+                a previous run. 'must' enforces the run must resume from the run with same id and throws an error
+                if it does not exist. 'never' enforces that a run will not resume and throws an error if run id exists.
+                'allow' resumes if the run id already exists. Defaults to 'allow'.
+            kwargs (Any): Keyword arguments to wandb.init excluding the ones explicitly described above.
                 Documentation here: https://docs.wandb.ai/ref/python/init/
         """
 
@@ -77,6 +82,7 @@ def __init__(
         self.tags = tags
         self.name = name
         self.id = id
+        self.resume = resume
 
         # Keep track of epoch and step. Initialize as 0.
         self.current_epoch = 0
@@ -157,6 +163,7 @@ def start_run(self, wandb_init_kwargs: dict[str, Any]) -> None:
             tags=self.tags,
             name=self.name,
             id=self.id,
+            resume=self.resume,
             **wandb_init_kwargs,  # Other less commonly used kwargs
         )
         self.run_id = self.run._run_id  # If run_id was None, we need to reset run id