chore: version compatibility for 0.3.0

AnswerDotAI · May 17, 2024 · 835b7a8 · 835b7a8
1 parent 01febd3
commit 835b7a8
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -61,6 +61,7 @@ pip install "rerankers[api]"
 pip install "rerankers[fastrank]"
 
 # RankLLM rerankers (better RankGPT + support for local models such as RankZephyr and RankVicuna)
+# Note: RankLLM is only supported on Python 3.10+! This will not work with Python 3.9
 pip install "rerankers[rankllm]"
 
 # All of the above

diff --git a/pyproject.toml b/pyproject.toml
@@ -52,8 +52,17 @@ dependencies = [
 ]
 
 [project.optional-dependencies]
-all = ["transformers", "torch", "litellm", "requests", "sentencepiece", "protobuf", "flashrank", "rank-llm"]
-transformers = ["transformers", "torch", "sentencepiece",  "protobuf"]
+all = [
+    "transformers",
+    "torch",
+    "litellm",
+    "requests",
+    "sentencepiece",
+    "protobuf",
+    "flashrank",
+    "rank-llm; python_version >= '3.10'"
+]
+transformers = ["transformers", "torch", "sentencepiece", "protobuf"]
 api = ["requests"]
 gpt = ["litellm"]
 flashrank = ["flashrank"]