Skip to content

Commit

Permalink
chore: minor deps fix (colbert-ai no longer required)
Browse files Browse the repository at this point in the history
  • Loading branch information
bclavie committed Mar 15, 2024
1 parent aa95135 commit cefcbda
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ packages = [
name = "rerankers"


version = "0.1.0"
version = "0.1.0post1"

description = "A unified API for various document re-ranking models."

Expand Down Expand Up @@ -52,7 +52,7 @@ dependencies = [
]

[project.optional-dependencies]
all = ["transformers", "torch", "litellm", "requests", "sentencepiece", "protobuf", "colbert-ai"]
all = ["transformers", "torch", "litellm", "requests", "sentencepiece", "protobuf"]
transformers = ["transformers", "torch", "sentencepiece", "protobuf"]
api = ["requests"]
gpt = ["litellm"]
Expand Down
2 changes: 1 addition & 1 deletion rerankers/__init__.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from rerankers.reranker import Reranker

__all__ = ["Reranker"]
__version__ = "0.1.0"
__version__ = "0.1.0post1"

0 comments on commit cefcbda

Please sign in to comment.