as node mask for as per node data cls

aditya0by0 · aditya0by0 · commit d0ea81d9f67d · 2025-08-11T18:48:53.000+02:00
diff --git a/chebai_graph/preprocessing/datasets/chebi.py b/chebai_graph/preprocessing/datasets/chebi.py
@@ -510,6 +510,9 @@ def _merge_props_into_base(
             edge_index=geom_data.edge_index,
             edge_attr=edge_attr,
             molecule_attr=torch.empty((1, 0)),  # empty as not used for this class
+            is_atom_node=is_atom_node,
+            is_fg_node=is_fg_node,
+            is_graph_node=is_graph_node,
         )
 
 

Original file line number	Diff line number	Diff line change
`@@ -510,6 +510,9 @@ def _merge_props_into_base(`
`510`	`510`	`edge_index=geom_data.edge_index,`
`511`	`511`	`edge_attr=edge_attr,`
`512`	`512`	`molecule_attr=torch.empty((1, 0)), # empty as not used for this class`
	`513`	`+ is_atom_node=is_atom_node,`
	`514`	`+ is_fg_node=is_fg_node,`
	`515`	`+ is_graph_node=is_graph_node,`
`513`	`516`	`)`
`514`	`517`
`515`	`518`