fix GAT model edge-index and act error

aditya0by0 · aditya0by0 · commit 7e074f5fe1b9 · 2025-07-08T16:39:18.000+02:00
diff --git a/chebai_graph/models/gat.py b/chebai_graph/models/gat.py
@@ -37,7 +37,7 @@ def __init__(self, config: dict, **kwargs):
             edge_dim=self.n_bond_properties,
             heads=self.heads,
             v2=self.v2,
-            act=ELU,
+            act=self.activation,
         )
 
     def forward(self, batch: dict) -> torch.Tensor:
@@ -59,7 +59,7 @@ def forward(self, batch: dict) -> torch.Tensor:
 
         out = self.gat(
             x=graph_data.x.float(),
-            edge_index=graph_data.edge_index,
+            edge_index=graph_data.edge_index.long(),
             edge_attr=graph_data.edge_attr,
         )
 

Original file line number	Diff line number	Diff line change
`@@ -37,7 +37,7 @@ def __init__(self, config: dict, **kwargs):`
`37`	`37`	`edge_dim=self.n_bond_properties,`
`38`	`38`	`heads=self.heads,`
`39`	`39`	`v2=self.v2,`
`40`		`- act=ELU,`
	`40`	`+ act=self.activation,`
`41`	`41`	`)`
`42`	`42`
`43`	`43`	`def forward(self, batch: dict) -> torch.Tensor:`
`@@ -59,7 +59,7 @@ def forward(self, batch: dict) -> torch.Tensor:`
`59`	`59`
`60`	`60`	`out = self.gat(`
`61`	`61`	`x=graph_data.x.float(),`
`62`		`- edge_index=graph_data.edge_index,`
	`62`	`+ edge_index=graph_data.edge_index.long(),`
`63`	`63`	`edge_attr=graph_data.edge_attr,`
`64`	`64`	`)`
`65`	`65`