Make check more robust (issues #10 and #11)

lmcinnes · Nov 13, 2017 · 5c52e74 · 5c52e74
1 parent a279df6
commit 5c52e74
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/umap/umap_.py b/umap/umap_.py
@@ -360,7 +360,7 @@ def fuzzy_simplicial_set(X, n_neighbors, metric, metric_kwds={}):
                                                leaf_array=leaf_array)
     knn_indices = tmp_indices.astype(np.int64)
 
-    if np.any(knn_indices == -1):
+    if np.any(knn_indices < 0):
         warn('Failed to correctly find n_neighbors for some samples.'
              'Results may be less than ideal. Try re-running with'
              'different parameters.')