Issue #71 touch up

lmcinnes · May 24, 2018 · c86884d · c86884d
1 parent 9e199e4
commit c86884d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/umap/umap_.py b/umap/umap_.py
@@ -1408,7 +1408,7 @@ def fit(self, X, y=None):
             print("Construct fuzzy simplicial set")
 
         # Handle small cases efficiently by computing all distances
-        if X.shape[0] < 4096 and not (metric == 'correlation' and scipy.sparse.isspmatrix_csr(X)):
+        if X.shape[0] < 4096 and not (self.metric == 'correlation' and scipy.sparse.isspmatrix_csr(X)):
             dmat = pairwise_distances(X, metric=self.metric, **self.metric_kwds)
             self.graph = fuzzy_simplicial_set(
                 dmat,