diff --git a/setup.py b/setup.py index 410983f..97a53e3 100644 --- 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",