From f3c09399ea000b2aea4bf2c7f30731d651ea367e Mon Sep 17 00:00:00 2001
From: Ilyes Batatia <48651863+ilyes319@users.noreply.github.com>
Date: Fri, 22 Nov 2024 15:57:09 +0000
Subject: [PATCH] fix interaction_first for density

---
 mace/tools/model_script_utils.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/mace/tools/model_script_utils.py b/mace/tools/model_script_utils.py
index ad54e2f0..d937446c 100644
--- a/mace/tools/model_script_utils.py
+++ b/mace/tools/model_script_utils.py
@@ -146,6 +146,11 @@ def _build_model(
     args, model_config, model_config_foundation, heads
 ):  # pylint: disable=too-many-return-statements
     if args.model == "MACE":
+        if args.interaction_first not in [
+            "RealAgnosticInteractionBlock",
+            "RealAgnosticDensityInteractionBlock",
+        ]:
+            args.interaction_first = "RealAgnosticInteractionBlock"
         return modules.ScaleShiftMACE(
             **model_config,
             pair_repulsion=args.pair_repulsion,