Merge pull request #1300 from DrDavidS/master

[Bugfix] Typehint error on py38
FlagOpen · Dec 30, 2024 · ff3c36a · ff3c36a
2 parents bae5fab + b83f16a
commit ff3c36a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/FlagEmbedding/abc/finetune/reranker/AbsDataset.py b/FlagEmbedding/abc/finetune/reranker/AbsDataset.py
@@ -183,7 +183,7 @@ class AbsRerankerCollator(DataCollatorWithPadding):
     query_max_len: int = 32
     passage_max_len: int = 128
 
-    def __call__(self, features) -> list[BatchEncoding]:
+    def __call__(self, features) -> List[BatchEncoding]:
         teacher_scores = [f[1] for f in features]
         if teacher_scores[0] is None:
             teacher_scores = None