Feature: add wandb dir as parser argument (#308)

* feat: add wandb_dir as cfg argument

* refactor: match functional dir style for wandb dir

sample_factory/cfg/cfg.py

@@ -744,6 +744,12 @@ def add_wandb_args(p: ArgumentParser):
         nargs="*",
         help="Tags can help with finding experiments in WandB web console",
     )
+    p.add_argument(
+        "--wandb_dir",
+        default=join(os.getcwd(), "wandb"),
+        type=str,
+        help="Logging Directory for WandB",
+    )
 
 
 def add_pbt_args(p: ArgumentParser):

sample_factory/utils/utils.py

@@ -402,6 +402,10 @@ def experiments_dir(cfg, mkdir=True) -> str:
     return maybe_ensure_dir_exists(cfg.train_dir, mkdir)
 
 
+def wandb_dir(cfg, mkdir=True) -> str:
+    return maybe_ensure_dir_exists(cfg.wandb_dir, mkdir)
+
+
 def experiment_dir(cfg, mkdir=True) -> str:
     experiment = cfg.experiment
     experiments_root = experiments_dir(cfg, mkdir)

sample_factory/utils/wandb_utils.py

@@ -1,6 +1,6 @@
 from datetime import datetime
 
-from sample_factory.utils.utils import log, retry
+from sample_factory.utils.utils import log, retry, wandb_dir
 
 
 def init_wandb(cfg):
@@ -44,6 +44,7 @@ def init_wandb_func():
             tags=cfg.wandb_tags,
             resume="allow",
             settings=wandb.Settings(start_method="fork"),
+            dir=wandb_dir(cfg, True),
         )
 
     log.debug("Initializing WandB...")