fix: fix type hints

qdrant · Oct 25, 2024 · f7986de · f7986de
1 parent 3a0c8eb
commit f7986de
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/qdrant_client/hybrid/test_reranking.py b/qdrant_client/hybrid/test_reranking.py
@@ -1,3 +1,5 @@
+from typing import List
+
 import numpy as np
 
 from qdrant_client.http import models
@@ -52,7 +54,7 @@ def test_distribution_based_score_fusion() -> None:
 
 
 def test_reciprocal_rank_fusion_empty_responses() -> None:
-    responses = [[]]
+    responses: List[List[models.ScoredPoint]] = [[]]
     fused = reciprocal_rank_fusion(responses)
     assert fused == []
 
@@ -76,7 +78,7 @@ def test_reciprocal_rank_fusion_empty_responses() -> None:
 
 
 def test_distribution_based_score_fusion_empty_response() -> None:
-    responses = [[]]
+    responses: List[List[models.ScoredPoint]] = [[]]
     fused = distribution_based_score_fusion(responses, limit=3)
     assert fused == []