nngraph: fix attributes

epfl-lts2 · Feb 24, 2019 · bfef548 · bfef548
1 parent 9c8e86e
commit bfef548
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/pygsp/graphs/nngraphs/nngraph.py b/pygsp/graphs/nngraphs/nngraph.py
@@ -322,7 +322,7 @@ def __init__(self, features, standardize=False,
                 radius = graph.kernel_width
             elif radius <= 0:
                 raise ValueError('The radius must be greater than 0.')
-            k = None
+            self.k = None
         else:
             raise ValueError('Invalid kind "{}".'.format(kind))
 

diff --git a/pygsp/tests/test_graphs.py b/pygsp/tests/test_graphs.py
@@ -448,6 +448,10 @@ def test_nngraph(self, n_vertices=24):
         Graph(data, kernel=lambda d: d.min()/d)
         self.assertRaises(ValueError, Graph, data, kernel=lambda d: 1/d)
 
+        # Attributes.
+        self.assertEqual(Graph(data, kind='knn').radius, None)
+        self.assertEqual(Graph(data, kind='radius').k, None)
+
     def test_bunny(self):
         graphs.Bunny()