Use nmslib from pypi

scikit-learn-contrib · Jun 29, 2023 · ad7ab2e · ad7ab2e
1 parent 7140097
commit ad7ab2e
Show file tree

Hide file tree

Showing 3 changed files with 96 additions and 4 deletions.
diff --git a/install_all.sh b/install_all.sh
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -8,6 +8,7 @@ license = "BSD"
 [tool.poetry.dependencies]
 python = "<3.13,>=3.9"
 scikit-learn = ">=0.24.0"
+scipy = "^1.11.1"
 pytest = {version = "^6.2.1", optional = true}
 pytest-cov = {version = "^2.10.1", optional = true}
 sphinx = {version = "^3.4.1", optional = true}
@@ -22,7 +23,7 @@ black = {version = "=19.10b0", optional = true}
 pynndescent = {version = "^0.5.1", optional = true}
 pandas = {version = "^1.2.0", optional = true, python = "^3.7.1"}
 sphinx-issues = {version = "^1.2.0", optional = true}
-scipy = "^1.11.1"
+nmslib = {version = "^2.1.1", optional = true}
 
 [tool.poetry.dev-dependencies]
 
@@ -36,3 +37,4 @@ docs = ["sphinx", "sphinx-gallery", "sphinx_rtd_theme", "sphinx-issues", "numpyd
 annoy = ["annoy"]
 faiss = ["faiss-cpu"]
 pynndescent = ["pynndescent"]
+nmslib = ["nmslib"]