Added umap-learn to dependencies until it can be put as a conditional…

… import.
ejohnson643 · Dec 28, 2021 · 20d3a6e · 20d3a6e
1 parent 103a21f
commit 20d3a6e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -260,7 +260,8 @@ def build_extensions(self):
         "scikit-learn>=0.20",
         "scipy",
         "cython",
-        "numba"
+        "numba",
+        "umap-learn"
     ],
     extras_require={
         "pynndescent": "pynndescent~=0.5.0",