From 989eaf2db34157f4f84ce0cd2b4b359ebd13ca42 Mon Sep 17 00:00:00 2001
From: allaffa <massimiliano.lupo.pasini@gmail.com>
Date: Tue, 26 Nov 2024 00:07:53 -0500
Subject: [PATCH] edge_dim passed into GATv2Conv stack inside create method

---
 hydragnn/models/create.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/hydragnn/models/create.py b/hydragnn/models/create.py
index 7c6949d5..ef24a3a9 100644
--- 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,