fix knn balltree warning

KrishnaswamyLab · Jun 19, 2018 · 2f312c9 · 2f312c9
1 parent ee767f9
commit 2f312c9
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/graphtools/graphs.py b/graphtools/graphs.py
@@ -154,9 +154,11 @@ def knn_tree(self):
                     n_jobs=self.n_jobs).fit(self.data_nu)
             except ValueError:
                 # invalid metric
-                log_warning(
+                warnings.warn(
                     "Metric {} not valid for `sklearn.neighbors.BallTree`. "
-                    "Graph instantiation may be slower than normal.")
+                    "Graph instantiation may be slower than normal.".format(
+                        self.distance),
+                    UserWarning)
                 self._knn_tree = NearestNeighbors(
                     n_neighbors=self.knn,
                     algorithm='auto',