From ff14d0b1201cf58e603ecdefe286a7f7844226bc Mon Sep 17 00:00:00 2001
From: Theo Keane <20931519+theochemtheo@users.noreply.github.com>
Date: Tue, 3 Mar 2026 10:28:33 +0000
Subject: [PATCH] permit first block to be non-linear in 'MACE'-type models

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

diff --git a/mace/tools/model_script_utils.py b/mace/tools/model_script_utils.py
index 53c71eaa6..76bfc9b72 100644
--- a/mace/tools/model_script_utils.py
+++ b/mace/tools/model_script_utils.py
@@ -244,6 +244,7 @@ def _build_model(
         if args.interaction_first not in [
             "RealAgnosticInteractionBlock",
             "RealAgnosticDensityInteractionBlock",
+            "RealAgnosticResidualNonLinearInteractionBlock",
         ]:
             args.interaction_first = "RealAgnosticInteractionBlock"
         return modules.ScaleShiftMACE(