1
0

base_reranker.py 219 B

12345678
  1. from abc import ABC, abstractmethod
  2. from typing import Optional, List, Tuple
  3. class BaseReranker(ABC):
  4. @abstractmethod
  5. def predict(self, sentences: List[Tuple[str, str]]) -> Optional[List[float]]:
  6. pass