test with multiple workers

pyg-team · rusty1s · Aug 10, 2022 · Aug 4, 2022 · Aug 4, 2022 · Aug 4, 2022
commit 853bf2f2717f4eaab0da74f8637f10a0a2faa7e6
@@ -206,7 +206,8 @@ def test_temporal_heterogeneous_link_neighbor_loader():
     loader = LinkNeighborLoader(data, num_neighbors=[-1] * 2,
                                 edge_label_index=('paper', 'paper'),
                                 edge_label_time=edge_time, batch_size=32,
-                                time_attr='time', neg_sampling_ratio=0.5)
+                                time_attr='time', neg_sampling_ratio=0.5,
+                                num_workers=2)
     for batch in loader:
         author_max = batch['author'].time.max()
         edge_max = batch['paper', 'paper'].edge_label_time.max()