radu.mutilica commited on
Commit
d5609b4
·
1 Parent(s): 0de9240

logging and pydantic issues

Browse files
Files changed (1) hide show
  1. handler.py +3 -2
handler.py CHANGED
@@ -48,15 +48,16 @@ class EndpointHandler:
48
  f'Loaded {path}:top_k={self.top_k}:length={self.max_length}:workers={self.max_workers}')
49
 
50
  def __call__(self, data: Dict[str, str]) -> List[Rank]:
51
- inputs = data.pop('inputs')
52
-
53
  logger.info(f'Received new docs to rerank: {data}')
 
 
54
  ranks = self.model.rank(
55
  inputs['query'],
56
  inputs['documents'],
57
  top_k=self.top_k,
58
  num_workers=self.max_workers
59
  )
 
60
 
61
  ranks = [Rank(
62
  corpus_id=r['corpus_id'],
 
48
  f'Loaded {path}:top_k={self.top_k}:length={self.max_length}:workers={self.max_workers}')
49
 
50
  def __call__(self, data: Dict[str, str]) -> List[Rank]:
 
 
51
  logger.info(f'Received new docs to rerank: {data}')
52
+
53
+ inputs = data.pop('inputs')
54
  ranks = self.model.rank(
55
  inputs['query'],
56
  inputs['documents'],
57
  top_k=self.top_k,
58
  num_workers=self.max_workers
59
  )
60
+ logger.info(f'New ranks: {ranks}')
61
 
62
  ranks = [Rank(
63
  corpus_id=r['corpus_id'],