backwards support

FAIR-Chem · May 28, 2024 · 73488f9 · 73488f9
1 parent d12b514
commit 73488f9
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/fairchem/core/common/logger.py b/src/fairchem/core/common/logger.py
@@ -61,14 +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=self.config["logger"].get("entity", None),
+            entity=entity,
             resume="allow",
         )