File size: 219 Bytes
87bf6ae
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
from abc import ABC, abstractmethod
from typing import Optional, List, Tuple


class BaseReranker(ABC):
    @abstractmethod
    def predict(self, sentences: List[Tuple[str, str]]) -> Optional[List[float]]:
        pass