Merge branch 'modernbert'

StarlightSearch · Jan 5, 2025 · 4a904fb · 4a904fb
2 parents 7d458ce + 5c8b3e1
commit 4a904fb
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 4 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/python/Cargo.toml b/python/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "embed_anything_python"
-version = "0.4.18"
+version = "0.5.0"
 edition = "2021"
 
 [lib]

diff --git a/python/python/embed_anything/_embed_anything.pyi b/python/python/embed_anything/_embed_anything.pyi
@@ -336,6 +336,9 @@ class Reranker:
         """
 
 class Dtype(Enum):
+    """
+    Represents the data type of the model.
+    """
     F16 = "F16"
     INT8 = "INT8"
     Q4 = "Q4"
@@ -345,18 +348,28 @@ class Dtype(Enum):
 class RerankerResult:
     """
     Represents the result of the reranking process.
+
+    Attributes:
+        query: The query to rerank.
+        documents: The list of documents to rerank.
     """
     query: str
     documents: list[DocumentRank]
 
 class DocumentRank:
     """
     Represents the rank of a document.
+
+    Attributes:
+        document: The document to rank.
+        relevance_score: The relevance score of the document.
+        rank: The rank of the document.
     """
     document: str
     relevance_score: float
     rank: int
 
+
 class TextEmbedConfig:
     """
     Represents the configuration for the Text Embedding model.