Workaround to WandbLogger ignoring WANDB_DIR

Author: lucmos

src/nn_core/model_logging.py

@@ -39,7 +39,11 @@ def __init__(self, logging_cfg: DictConfig, cfg: DictConfig, resume_id: Optional
             self.logging_cfg.logger.mode = "offline"
 
         pylogger.info(f"Instantiating <{self.logging_cfg.logger['_target_'].split('.')[-1]}>")
-        self.wrapped: LightningLoggerBase = hydra.utils.instantiate(self.logging_cfg.logger, version=self.resume_id)
+        self.wrapped: LightningLoggerBase = hydra.utils.instantiate(
+            self.logging_cfg.logger,
+            version=self.resume_id,
+            dir=os.getenv("WANDB_DIR", "."),
+        )
 
         # force experiment lazy initialization
         _ = self.wrapped.experiment