fix doc

epfl-lts2 · Aug 19, 2019 · 3ed33f9 · 3ed33f9
1 parent b3c8efe
commit 3ed33f9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/pygsp/graphs/learngraph.py b/pygsp/graphs/learngraph.py
@@ -509,7 +509,7 @@ class LearnGraph(Graph):
     >>> 
     >>> # B) Learn the graph
     >>> param_opt = {'verbosity':0}
-    >>> Glearned = LearnGraph(X,k=k, param_opt=param_opt)
+    >>> Glearned = pg.graphs.LearnGraph(X,k=k, param_opt=param_opt)
     >>> # plot the learned graph
     >>> Glearned.coords = coords
     >>> 

diff --git a/pygsp/tests/test_learning.py b/pygsp/tests/test_learning.py
@@ -132,3 +132,6 @@ def test_classification_tikhonov(self):
 
 
 suite = unittest.TestLoader().loadTestsFromTestCase(TestCase)
+
+if __name__ == '__main__':
+    unittest.main()