Update main.py (#5460)

examples/pytorch/ogb/seal_ogbl/main.py

@@ -463,12 +463,12 @@ if __name__ == "__main__":
 
     # reconstruct the graph for ogbl-collab data for validation edge augmentation and coalesce
     if args.dataset == "ogbl-collab":
+        graph.edata.pop("year")
+        # float edata for to_simple transform
+        graph.edata["weight"] = graph.edata["weight"].to(torch.float)
         if args.use_valedges_as_input:
             val_edges = split_edge["valid"]["edge"]
             row, col = val_edges.t()
-            # float edata for to_simple transform
-            graph.edata.pop("year")
-            graph.edata["weight"] = graph.edata["weight"].to(torch.float)
             val_weights = torch.ones(size=(val_edges.size(0), 1))
             graph.add_edges(
                 torch.cat([row, col]),