Merge branch 'master' into link_time_attr

torch_geometric/loader/link_neighbor_loader.py

@@ -340,13 +340,13 @@ def __init__(
         edge_type, edge_label_index = get_edge_label_index(
             data, edge_label_index)
         if edge_label is None:
-            edge_label = edge_label_index.new_zeros(edge_label_index.size(1))
-            edge_label = (edge_label +
-                          1 if self.neg_sampling_ratio == 0 else edge_label)
-        self.edge_label = edge_label.to(torch.float)
+            edge_label = torch.zeros(edge_label_index.size(1),
+                                     device=edge_label_index.device)
+
         if edge_time is not None and time_attr is None:
             raise ValueError("`time_attr` has to be specified if"
                              "`edge_time` is set")
+
         if neighbor_sampler is None:
             self.neighbor_sampler = LinkNeighborSampler(
                 data,