We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents b06a117 + fd96416 commit 2fd437fCopy full SHA for 2fd437f
torch_geometric/loader/neighbor_loader.py
@@ -108,7 +108,8 @@ def __init__(
108
node_attrs = feature_store.get_all_tensor_attrs()
109
edge_attrs = graph_store.get_all_edge_attrs()
110
111
- self.node_types = list(set(node_attr.group_name for node_attr in node_attrs))
+ self.node_types = list(
112
+ set(node_attr.group_name for node_attr in node_attrs))
113
self.edge_types = [edge_attr.edge_type for edge_attr in edge_attrs]
114
115
# Set other required parameters:
0 commit comments