install hydra and torchtnt (#967)

FAIR-Chem · Jan 15, 2025 · 4f42b67 · 4f42b67
1 parent f290b21
commit 4f42b67
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/packages/env.gpu.yml b/packages/env.gpu.yml
@@ -21,6 +21,8 @@ dependencies:
     - torch_scatter==2.1.2+pt24cu121
     - torch_sparse==0.6.18+pt24cu121
     - torch_spline_conv==1.2.2+pt24cu121
+    - hydra-core==1.3.2
+    - torchtnt==0.2.4
 - pyyaml
 - tqdm
 - python-lmdb

diff --git a/src/fairchem/core/common/utils.py b/src/fairchem/core/common/utils.py
@@ -975,7 +975,7 @@ def setup_logging() -> None:
     root.setLevel(target_logging_level)
     if not root.hasHandlers():
         log_formatter = logging.Formatter(
-            "%(asctime)s (%(levelname)s): %(message)s",
+            "%(asctime)s %(pathname)s:%(lineno)d: (%(levelname)s): %(message)s",
             datefmt="%Y-%m-%d %H:%M:%S",
         )