Fix dataframe not being recognised (issue #23)

lmcinnes · Nov 21, 2017 · e2abc3a · e2abc3a
1 parent 411dda4
commit e2abc3a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/umap/umap_.py b/umap/umap_.py
@@ -6,7 +6,7 @@
 
 from scipy.optimize import curve_fit
 from sklearn.base import BaseEstimator
-from sklearn.utils import check_random_state
+from sklearn.utils import check_random_state, check_array
 
 import numpy as np
 import scipy.sparse
@@ -1520,7 +1520,7 @@ def fit(self, X, y=None):
         """
 
         # Handle other array dtypes (TODO: do this properly)
-        X = X.astype(np.float64)
+        X = check_array(X).astype(np.float64)
 
         random_state = check_random_state(self.random_state)