Merge pull request #237 from jamesmyatt/bugfix/joblib_import

Fix deprecation warning with scikit-learn 0.21
lmcinnes · May 16, 2019 · 7041502 · 7041502
2 parents b381667 + 3b96697
commit 7041502
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/umap/umap_.py b/umap/umap_.py
@@ -12,7 +12,11 @@
 from sklearn.preprocessing import normalize
 from sklearn.neighbors import KDTree
 
-from sklearn.externals import joblib
+try:
+    import joblib
+except ImportError:
+    # sklearn.externals.joblib is deprecated in 0.21, will be removed in 0.23
+    from sklearn.externals import joblib
 
 import numpy as np
 import scipy.sparse