[Patch] Respect ALF_WANDB_ENTITY

HorizonRobotics · Jun 17, 2024 · 2a32fe9 · 2a32fe9
1 parent ad586df
commit 2a32fe9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/alf/bin/train.py b/alf/bin/train.py
@@ -155,7 +155,7 @@ def _train(conf_file: str, root_dir, rank=0, world_size=1):
 
         wandb.init(
             project=project,
-            entity="horizon-robotics-gail",
+            entity=os.getenv("ALF_WANDB_ENTITY", "horizon-robotics-gail"),
             group=group,
             name=name,
             reinit=True,