fix array comparison

KrishnaswamyLab · Jun 30, 2018 · e5e2d28 · e5e2d28
1 parent 14a1973
commit e5e2d28
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/magic/magic.py b/python/magic/magic.py
@@ -489,7 +489,7 @@ def transform(self, X=None, genes=None, t_max=20,
                 self.X_magic = X_magic
 
         # return selected genes
-        if genes == "pca_only":
+        if isinstance(genes, str) and genes == "pca_only":
             if sparse.issparse(self.graph.data):
                 X_magic = PCA().fit_transform(X_magic)
             genes = ["PC{}".format(i + 1) for i in range(X_magic.shape[1])]