fix transform after set_params

KrishnaswamyLab · Jun 29, 2018 · d1c281d · d1c281d
1 parent e9f3829
commit d1c281d
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/python/magic/magic.py b/python/magic/magic.py
@@ -429,9 +429,13 @@ def transform(self, X=None, genes=None, t_max=20,
             pass
 
         if self.graph is None:
-            raise NotFittedError("This MAGIC instance is not fitted yet. Call "
-                                 "'fit' with appropriate arguments before "
-                                 "using this method.")
+            if self.X is not None:
+                self.fit(self.X)
+            else:
+                raise NotFittedError(
+                    "This MAGIC instance is not fitted yet. Call "
+                    "'fit' with appropriate arguments before "
+                    "using this method.")
 
         store_result = True
         if X is not None and not matrix_is_equivalent(X, self.X):