fixes

Author: CarloLucibello

docs/make.jl

@@ -19,6 +19,7 @@ makedocs(
     ),
 
     authors = "Hiroyuki Shindo, Christof Stocker",
+    # TODO: automatize `pages` creation
     pages = Any[
         "Home" => "index.md",
         "Available Datasets" => Any[
@@ -40,10 +41,13 @@ makedocs(
             ],
 
             "Graphs" => Any[
+                "CiteSeer" => "datasets/CiteSeer.md",
                 "Cora" => "datasets/Cora.md",
+                "PubMed" => "datasets/PubMed.md",
             ],
 
         ],
+        "Utis" -> "utils.md",
         "LICENSE.md",
     ],
     strict = true

docs/src/utils.md

@@ -0,0 +1,4 @@
+# Utils
+```@docs
+MLDatasets.read_planetoid_data
+```

test/tst_citeseer.jl

@@ -7,6 +7,7 @@ end
 
     @test data.num_nodes == 3327
     @test data.num_edges == 9104
+    @test data.directed == false
     @test size(data.node_features) == (3703, data.num_nodes)
     @test size(data.node_labels) == (data.num_nodes,)
     @test size(data.train_indices) == (120,)

test/tst_cora.jl

@@ -7,7 +7,7 @@ end
 
     @test data.num_nodes == 2708
     @test data.num_edges == 10556
-    @test data.directed == true
+    @test data.directed == false
     @test size(data.node_features) == (1433, data.num_nodes)
     @test size(data.node_labels) == (data.num_nodes,)
     @test size(data.train_indices) == (140,)

test/tst_pubmed.jl

@@ -7,7 +7,7 @@ end
 
     @test data.num_nodes == 19717
     @test data.num_edges == 88648
-    @test data.directed == true
+    @test data.directed == false
     @test size(data.node_features) == (500, data.num_nodes)
     @test size(data.node_labels) == (data.num_nodes,)
     @test size(data.train_indices) == (60,)