Skip to content

Commit

Permalink
qdrant - set scale_score to False
Browse files Browse the repository at this point in the history
  • Loading branch information
anakin87 committed Jul 1, 2024
1 parent b051617 commit c84fa48
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ def __init__(
document_store: QdrantDocumentStore,
filters: Optional[Union[Dict[str, Any], models.Filter]] = None,
top_k: int = 10,
scale_score: bool = True,
scale_score: bool = False,
return_embedding: bool = False,
):
"""
Expand Down Expand Up @@ -159,7 +159,7 @@ def __init__(
document_store: QdrantDocumentStore,
filters: Optional[Union[Dict[str, Any], models.Filter]] = None,
top_k: int = 10,
scale_score: bool = True,
scale_score: bool = False,
return_embedding: bool = False,
):
"""
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -496,7 +496,7 @@ def _query_by_sparse(
query_sparse_embedding: SparseEmbedding,
filters: Optional[Union[Dict[str, Any], rest.Filter]] = None,
top_k: int = 10,
scale_score: bool = True,
scale_score: bool = False,
return_embedding: bool = False,
) -> List[Document]:
"""
Expand Down Expand Up @@ -553,7 +553,7 @@ def _query_by_embedding(
query_embedding: List[float],
filters: Optional[Union[Dict[str, Any], rest.Filter]] = None,
top_k: int = 10,
scale_score: bool = True,
scale_score: bool = False,
return_embedding: bool = False,
) -> List[Document]:
"""
Expand Down
4 changes: 2 additions & 2 deletions integrations/qdrant/tests/test_retriever.py
Original file line number Diff line number Diff line change
Expand Up @@ -71,7 +71,7 @@ def test_to_dict(self):
},
"filters": None,
"top_k": 10,
"scale_score": True,
"scale_score": False,
"return_embedding": False,
},
}
Expand Down Expand Up @@ -190,7 +190,7 @@ def test_to_dict(self):
},
"filters": None,
"top_k": 10,
"scale_score": True,
"scale_score": False,
"return_embedding": False,
},
}
Expand Down

0 comments on commit c84fa48

Please sign in to comment.