File size: 310 Bytes
b06298d 51a31d4 b06298d |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
from typing import Dict, List, Tuple
import numpy as np
RetrieveTypeResult = Dict[str, List[str]]
RetrieveTypeScores = np.ndarray
RetrieveType = Tuple[RetrieveTypeScores, RetrieveTypeResult]
class Retriever():
def retrieve(self, query: str, k: int) -> RetrieveType:
raise NotImplementedError()
|