From 9f9870d990eb140009e3948ace2b7343a4c33c2e Mon Sep 17 00:00:00 2001
From: Matthew Edwards <medwards2@rmn.com>
Date: Wed, 25 Jul 2018 10:26:32 -0500
Subject: [PATCH] Fix fit(), then transform() bug referenced in #89

---
 umap/umap_.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/umap/umap_.py b/umap/umap_.py
index 72753cd6..e5e7e832 100644
--- a/umap/umap_.py
+++ b/umap/umap_.py
@@ -1622,7 +1622,7 @@ def transform(self, X):
             head,
             tail,
             n_epochs,
-            X.shape[0],
+            graph.shape[1],
             epochs_per_sample,
             self._a,
             self._b,