dynamic gni fix

aditya0by0 · aditya0by0 · commit 69cf0a59d576 · 2025-10-06T23:17:05.000+02:00
diff --git a/chebai_graph/models/dynamic_gni.py b/chebai_graph/models/dynamic_gni.py
@@ -61,9 +61,9 @@ def forward(self, batch: dict[str, Any]) -> Tensor:
         )
 
         out = self.resgated(
-            x=graph_data.x.float(),
+            x=random_x.float(),
             edge_index=graph_data.edge_index.long(),
-            edge_attr=graph_data.edge_attr,
+            edge_attr=random_edge_attr.float(),
         )
 
         return self.activation(out)

Original file line number	Diff line number	Diff line change
`@@ -61,9 +61,9 @@ def forward(self, batch: dict[str, Any]) -> Tensor:`
`61`	`61`	`)`
`62`	`62`
`63`	`63`	`out = self.resgated(`
`64`		`- x=graph_data.x.float(),`
	`64`	`+ x=random_x.float(),`
`65`	`65`	`edge_index=graph_data.edge_index.long(),`
`66`		`- edge_attr=graph_data.edge_attr,`
	`66`	`+ edge_attr=random_edge_attr.float(),`
`67`	`67`	`)`
`68`	`68`
`69`	`69`	`return self.activation(out)`