edge_dim passed into GATv2Conv stack inside create method

ORNL · Nov 26, 2024 · 989eaf2 · 989eaf2
1 parent 25d3dba
commit 989eaf2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/hydragnn/models/create.py b/hydragnn/models/create.py
@@ -205,6 +205,7 @@ def create_model(
             "inv_node_feat, edge_index",
             heads,
             negative_slope,
+            edge_dim,
             input_dim,
             hidden_dim,
             output_dim,