diff --git a/pyproject.toml b/pyproject.toml index 947afa9..f250f96 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -8,14 +8,17 @@ readme = "README.md" [tool.poetry.dependencies] python = "^3.7.1" -torch = "^1.13.0" +torch = ">=1.13.0 <3.0" sentence-transformers = "^2.2.2" transformers = "^4.29.2" openai = "^0.27.7" rank-bm25 = "^0.2.2" spacy = "^3.5.3" -pysqlite-binary = "^0.5.0" nltk = "^3.8.1" +pysqlite-binary = { version = "^0.5.0", optional = true } + +[tool.poetry.extras] +pysqlite-binary = ["pysqlite-binary"] [build-system] requires = ["poetry-core"]