concat edge attr for undirected graph

aditya0by0 · aditya0by0 · commit 53a240ad1b47 · 2025-05-14T23:24:14.000+02:00
diff --git a/chebai_graph/preprocessing/datasets/chebi.py b/chebai_graph/preprocessing/datasets/chebi.py
@@ -168,7 +168,7 @@ def _merge_props_into_base(self, row):
         return GeomData(
             x=x,
             edge_index=geom_data.edge_index,
-            edge_attr=edge_attr,
+            edge_attr=torch.cat([edge_attr, edge_attr], dim=0),
             molecule_attr=molecule_attr,
         )
 

Original file line number	Diff line number	Diff line change
`@@ -168,7 +168,7 @@ def _merge_props_into_base(self, row):`
`168`	`168`	`return GeomData(`
`169`	`169`	`x=x,`
`170`	`170`	`edge_index=geom_data.edge_index,`
`171`		`- edge_attr=edge_attr,`
	`171`	`+ edge_attr=torch.cat([edge_attr, edge_attr], dim=0),`
`172`	`172`	`molecule_attr=molecule_attr,`
`173`	`173`	`)`
`174`	`174`