GAT added in edge_models

ORNL · Nov 26, 2024 · 20032a0 · 20032a0
1 parent c0df8b0
commit 20032a0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/hydragnn/utils/input_config_parsing/config_utils.py b/hydragnn/utils/input_config_parsing/config_utils.py
@@ -149,6 +149,7 @@ def update_config_equivariance(config):
 def update_config_edge_dim(config):
     config["edge_dim"] = None
     edge_models = [
+        "GAT",
         "PNA",
         "PNAPlus",
         "PNAEq",