support logger entity (#717)

* support entity * backwards support
FAIR-Chem · May 31, 2024 · aa0f598 · aa0f598
1 parent b21a950
commit aa0f598
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/fairchem/core/common/logger.py b/src/fairchem/core/common/logger.py
@@ -61,13 +61,19 @@ def __init__(self, config) -> None:
             if isinstance(self.config["logger"], dict)
             else None
         )
+        entity = (
+            self.config["logger"].get("entity", None)
+            if isinstance(self.config["logger"], dict)
+            else None
+        )
 
         wandb.init(
             config=self.config,
             id=self.config["cmd"]["timestamp_id"],
             name=self.config["cmd"]["identifier"],
             dir=self.config["cmd"]["logs_dir"],
             project=project,
+            entity=entity,
             resume="allow",
         )