fix the edge index (#226)

Author: Dsantra92

src/datasets/graphs/ogbdataset.jl

@@ -268,7 +268,7 @@ function read_ogb_graph(path, metadata)
                     graph[k] = nothing
                 else
                     x = v[:, (num_edge_accum + 1):(num_edge_accum + m)]
-                    graph[k] = [x; x]
+                    graph[k] = [x x]
                 end
             end
         else
@@ -374,7 +374,7 @@ function read_ogb_graph(path, metadata)
                     else
                         if metadata["add_inverse_edge"]
                             v = split_dict[key][k]
-                            split_dict[key][k] = [v; v]
+                            split_dict[key][k] = [v v]
                         end
                     end
                 end

test/datasets/graphs_no_ci.jl

@@ -198,6 +198,8 @@ end
                          d.graph_data.test_mask,
                          d.graph_data.val_mask,
                      ])) == length(d)
+    g = d[1].graphs
+    @test size(g.edge_data.features, 2) == g.num_edges
 end
 
 @testset "Reddit_full" begin
@@ -376,4 +378,4 @@ end
     @test g.num_snapshots == 27
     @test g.snapshots[1] isa MLDatasets.Graph
     @test length(g.snapshots[1].node_data) == 102
-end
+end