From dad25632ab74c78eea8ab4779e3c337271f11e09 Mon Sep 17 00:00:00 2001
From: Janosh Riebesell <janosh.riebesell@gmail.com>
Date: Tue, 21 Nov 2023 19:37:04 -0800
Subject: [PATCH] train_mace.py update pip install mace comment

---
 .gitignore                | 2 --
 models/mace/train_mace.py | 9 +++------
 2 files changed, 3 insertions(+), 8 deletions(-)

diff --git a/.gitignore b/.gitignore
index 2e1961a1..3d18774c 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,13 +1,11 @@
 # package install
 *.egg-info
 dist
-build
 
 # cache
 __pycache__
 
 # datasets
-*.json
 *.json.gz
 *.json.bz2
 *.csv.bz2
diff --git a/models/mace/train_mace.py b/models/mace/train_mace.py
index 8a149381..f13431f2 100644
--- a/models/mace/train_mace.py
+++ b/models/mace/train_mace.py
@@ -36,11 +36,8 @@
 __date__ = "2023-09-18"
 
 
-# Running this script requires merging the multi-gpu and develop branches of the MACE
-# repo. Presumably, the next MACE repo release will include these changes at which point
-# this script can be run by simply installing the latest MACE release. For now, run
-# pip install git+https://github.com/ACEsuit/mace@develop
-# pip install git+https://github.com/chiang-yuan/mace@mbd
+# This script requires installing MACE.
+# pip install git+https://github.com/ACEsuit/mace
 
 module_dir = os.path.dirname(__file__)
 
@@ -555,7 +552,7 @@ def main(**kwargs: Any) -> None:
 
     logging.info(model)
     logging.info(f"Number of parameters: {tools.count_parameters(model)}")
-    logging.info(f"Optimizer: {optimizer}")
+    logging.info(f"{optimizer=}")
 
     if args.wandb:
         logging.info("Using Weights and Biases for logging")