Skip to content

Commit 6fabdfd

Browse files
committed
[IO] fix tu_dataset
1 parent a715fef commit 6fabdfd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

gammagl/io/tu.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -115,6 +115,6 @@ def split(edge_index, batch, x=None, edge_attr=None, y=None):
115115
slices['y'] = node_slice
116116
else:
117117
slices['y'] = np.arange(0, batch[-1] + 2, dtype=np.int32)
118-
graph = Graph(x=x, edge_index=edge_index, edge_feat=edge_attr, y=y).tensor()
118+
graph = Graph(x=x, edge_index=edge_index, edge_attr=edge_attr, y=y).tensor()
119119
return graph, slices
120120

0 commit comments

Comments
 (0)