diff --git a/python/graphstorm/dataloading/dataset.py b/python/graphstorm/dataloading/dataset.py index 7b8b159ad4..b87deaa45c 100644 --- a/python/graphstorm/dataloading/dataset.py +++ b/python/graphstorm/dataloading/dataset.py @@ -295,9 +295,11 @@ def _check_node_feats(self, node_feat_field): "or a dictionary of list of strings, " \ f"but get {node_feat_field}") for feat_name in feat_names: - assert feat_name in g.nodes[ntype].data, \ - f"Warning. The feature \"{feat_name}\" " \ - f"does not exists for the node type \"{ntype}\"." + assert feat_name in g.nodes[ntype].data, ( + f"The feature \"{feat_name}\" " + f"does not exist for the node type \"{ntype}\"." + f"Data available for \"{ntype}\": {g.nodes[ntype].data.keys()}" + ) def has_node_feats(self, ntype): """ Test if the specified node type has features. diff --git a/training_scripts/gsgnn_np/papers100M_nc.yaml b/training_scripts/gsgnn_np/papers100M_nc.yaml index ab0ef7917b..539ab1f13b 100644 --- a/training_scripts/gsgnn_np/papers100M_nc.yaml +++ b/training_scripts/gsgnn_np/papers100M_nc.yaml @@ -47,3 +47,5 @@ gsf: label_field: "label" multilabel: false num_classes: 172 + node_feat_name: + - "paper:paper_feat,paper_year"