Merge remote-tracking branch 'origin/jacob/windnet' into jacob/windnet

openclimatefix · Feb 6, 2024 · 6249e83 · 6249e83
2 parents 8888f4d + 9bb5c96
commit 6249e83
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pvnet/training.py b/pvnet/training.py
@@ -105,9 +105,9 @@ def train(config: DictConfig) -> Optional[float]:
             if isinstance(callback, ModelCheckpoint):
                 # Need to call the .experiment property to initialise the logger
                 wandb_logger.experiment
-                callback.dirpath = "/".join(callback.dirpath.split("/")[:-1] + [
-                    wandb_logger.version
-                ])
+                callback.dirpath = "/".join(
+                    callback.dirpath.split("/")[:-1] + [wandb_logger.version]
+                )
                 # Also save model config here - this makes for easy model push to huggingface
                 os.makedirs(callback.dirpath, exist_ok=True)
                 OmegaConf.save(config.model, f"{callback.dirpath}/model_config.yaml")