fix arg parser

ACEsuit · Nov 17, 2024 · f37a447 · f37a447
1 parent 2e11c83
commit f37a447
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/mace/tools/arg_parser.py b/mace/tools/arg_parser.py
@@ -153,6 +153,8 @@ def build_default_arg_parser() -> argparse.ArgumentParser:
             "RealAgnosticResidualInteractionBlock",
             "RealAgnosticAttResidualInteractionBlock",
             "RealAgnosticInteractionBlock",
+            "RealAgnosticDensityResidualInteractionBlock",
+            "RealAgnosticDensityInteractionBlock"
         ],
     )
     parser.add_argument(
@@ -163,6 +165,8 @@ def build_default_arg_parser() -> argparse.ArgumentParser:
         choices=[
             "RealAgnosticResidualInteractionBlock",
             "RealAgnosticInteractionBlock",
+            "RealAgnosticDensityResidualInteractionBlock",
+            "RealAgnosticDensityInteractionBlock"
         ],
     )
     parser.add_argument(