Skip to content

Commit c9608f1

Browse files
authored
Fix LinkNeighborLoader within multiprocessing.spawn (#5978)
1 parent b09990e commit c9608f1

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

torch_geometric/loader/link_loader.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -100,8 +100,12 @@ def __init__(
100100
self.data = data
101101
self.edge_type = edge_type
102102
self.link_sampler = link_sampler
103-
self.input_data = InputData(edge_label_index[0], edge_label_index[1],
104-
edge_label, edge_label_time)
103+
self.input_data = InputData(
104+
edge_label_index[0].clone(),
105+
edge_label_index[1].clone(),
106+
edge_label,
107+
edge_label_time,
108+
)
105109
self.neg_sampling_ratio = neg_sampling_ratio
106110
self.transform = transform
107111
self.filter_per_worker = filter_per_worker

0 commit comments

Comments
 (0)