Skip to content

Commit 2fd437f

Browse files
committed
2 parents b06a117 + fd96416 commit 2fd437f

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

torch_geometric/loader/neighbor_loader.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,8 @@ def __init__(
108108
node_attrs = feature_store.get_all_tensor_attrs()
109109
edge_attrs = graph_store.get_all_edge_attrs()
110110

111-
self.node_types = list(set(node_attr.group_name for node_attr in node_attrs))
111+
self.node_types = list(
112+
set(node_attr.group_name for node_attr in node_attrs))
112113
self.edge_types = [edge_attr.edge_type for edge_attr in edge_attrs]
113114

114115
# Set other required parameters:

0 commit comments

Comments
 (0)