From c57b6dac8922b9d2a32e3790fe1b9bfc967ccb7d Mon Sep 17 00:00:00 2001
From: Ilyes Batatia <48651863+ilyes319@users.noreply.github.com>
Date: Tue, 12 Nov 2024 20:12:51 +0000
Subject: [PATCH] fix name in foundation head args

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

diff --git a/mace/tools/arg_parser.py b/mace/tools/arg_parser.py
index 6d8be783..cb4f8ac5 100644
--- a/mace/tools/arg_parser.py
+++ b/mace/tools/arg_parser.py
@@ -361,7 +361,7 @@ def build_default_arg_parser() -> argparse.ArgumentParser:
         default=True,
     )
     parser.add_argument(
-        "--foundation_model_head",
+        "--foundation_head",
         help="Name of the head to use for fine-tuning",
         type=str,
         default=None,