allow for graphs without knn_max set, closes #180

KrishnaswamyLab · Jan 15, 2020 · f52f13a · f52f13a
1 parent 82edb3e
commit f52f13a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/python/magic/magic.py b/python/magic/magic.py
@@ -460,10 +460,14 @@ def fit(self, X, graph=None):
                     graph = None
         else:
             self.knn = graph.knn
-            self.knn_max = graph.knn_max
             self.alpha = graph.decay
             self.n_pca = graph.n_pca
             self.knn_dist = graph.distance
+            try:
+                self.knn_max = graph.knn_max
+            except AttributeError:
+                # not all graphs have knn_max
+                self.knn_max = None
 
         self.X = X