in kbc/models.py [0:0]
def get_ranking( self, queries: torch.Tensor, filters: Dict[Tuple[int, int], List[int]], batch_size: int = 1000, chunk_size: int = -1