mypy typing

getzep · Aug 26, 2024 · 13aa9fa · 13aa9fa
1 parent d901d88
commit 13aa9fa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/graphiti_core/search/search_utils.py b/graphiti_core/search/search_utils.py
@@ -333,7 +333,7 @@ async def get_relevant_edges(
 
 # takes in a list of rankings of uuids
 def rrf(results: list[list[str]], rank_const=1) -> list[str]:
-    scores: dict[str, int] = defaultdict(int)
+    scores: dict[str, float] = defaultdict(int)
     for result in results:
         for i, uuid in enumerate(result):
             scores[uuid] += 1 / (i + rank_const)
@@ -349,7 +349,7 @@ def rrf(results: list[list[str]], rank_const=1) -> list[str]:
 async def node_distance_reranker(
         driver: AsyncDriver, results: list[list[str]], center_node_uuid: str
 ) -> list[str]:
-    scores: dict[str, int] = {}
+    scores: dict[str, float] = {}
 
     for result in results:
         for uuid in result: