Merge pull request #265 from allenai/gabi/wandb_bugfix

Update the name of the wandb callback

Author: gabrieltseng

rslp/lightning_cli.py

@@ -74,7 +74,7 @@ def get_cached_checkpoint(checkpoint_fname: UPath) -> str:
     return local_fname
 
 
-class SaveWandbRunIdCallback(Callback):
+class SaveWandbRunIdCallbackRSLP(Callback):
     """Callback to save the wandb run ID to GCS in case of resume."""
 
     def __init__(
@@ -84,7 +84,7 @@ def __init__(
         run_id: str | None,
         config_str: str | None,
     ) -> None:
-        """Create a new SaveWandbRunIdCallback.
+        """Create a new SaveWandbRunIdCallbackRSLP.
 
         Args:
             project_id: the project ID.
@@ -365,7 +365,7 @@ def before_instantiate_classes(self) -> None:
                         == "lightning.pytorch.callbacks.ModelCheckpoint"
                     ):
                         checkpoint_callback = existing_callback
-                    if existing_callback.class_path == "SaveWandbRunIdCallback":
+                    if existing_callback.class_path == "SaveWandbRunIdCallbackRSLP":
                         upload_wandb_callback = existing_callback
             else:
                 c.trainer.callbacks = []
@@ -393,7 +393,7 @@ def before_instantiate_classes(self) -> None:
                 )
                 upload_wandb_callback = jsonargparse.Namespace(
                     {
-                        "class_path": "SaveWandbRunIdCallback",
+                        "class_path": "SaveWandbRunIdCallbackRSLP",
                         "init_args": jsonargparse.Namespace(
                             {
                                 "project_id": c.rslp_project,