rust/lancedb
1
0
Fork 0
mirror of https://github.com/lancedb/lancedb.git synced 2026-01-03 18:32:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: update type hint and remove unused import (#1436)

Browse Source 
change typehint on `_invert_score` from `List[float]` to `float`. remove
unnecessary typing import
This commit is contained in:
forrestmckee
2024-07-09 03:26:45 -05:00
committed by GitHub
parent 58c2cd01a5
commit b4436e0804
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
python/python/lancedb/rerankers/linear_combination.py
View File

@@ -1,5 +1,3 @@
from typing import List
import pyarrow as pa
from .base import Reranker
@@ -112,6 +110,6 @@ class LinearCombinationReranker(Reranker):
# these scores represent distance
return 1 - (self.weight * score1 + (1 - self.weight) * score2)
def _invert_score(self, scores: List[float]):
# Invert the scores between relevance and distance
return 1 - scores
def _invert_score(self, score: float):
# Invert the score between relevance and distance
return 1 - score