From 29c99ae084063230cbb032d43c3dda8b01447ddd Mon Sep 17 00:00:00 2001
From: Ilyes Batatia <48651863+ilyes319@users.noreply.github.com>
Date: Mon, 4 Nov 2024 12:02:42 +0000
Subject: [PATCH] fix formatting

---
 mace/cli/run_train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mace/cli/run_train.py b/mace/cli/run_train.py
index 9b484d7f..7f1a5e74 100644
--- a/mace/cli/run_train.py
+++ b/mace/cli/run_train.py
@@ -364,7 +364,7 @@ def run(args: argparse.Namespace) -> None:
                         z_table_foundation.z_to_index(z)
                     ].item()
                     for z in z_table.zs
-                }         
+                }
             else:
                 atomic_energies_dict[head_config.head_name] = get_atomic_energies(head_config.E0s, None, head_config.z_table)
         else: