chore: mark diskann as optional

2025-07-19 22:24:44 -07:00
parent 19c1b182c3
commit e60fad8c73
2 changed files with 14 additions and 1 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -9,7 +9,6 @@ requires-python = ">=3.10"

 dependencies = [
    "leann-core",
-    "leann-backend-diskann",
    "leann-backend-hnsw",
    "numpy>=1.26.0",
    "torch",
@@ -48,6 +47,10 @@ dev = [
    "huggingface-hub>=0.20.0",
 ]

+diskann = [
+    "leann-backend-diskann",
+]
+
 [tool.setuptools]
 py-modules = []