check for n_pca=='auto'

KrishnaswamyLab · Oct 12, 2019 · 786179f · 786179f
1 parent 17e066a
commit 786179f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/graphtools/base.py b/graphtools/base.py
@@ -232,7 +232,7 @@ def _reduce_data(self):
         -------
         Reduced data matrix
         """
-        if self.n_pca is not None and self.n_pca < self.data.shape[1]:
+        if self.n_pca is not None and (self.n_pca == 'auto' or self.n_pca < self.data.shape[1]):
             tasklogger.log_start("PCA")
             n_pca = self.data.shape[1] - 1 if self.n_pca == 'auto' else self.n_pca
             if sparse.issparse(self.data):