### What problem does this PR solve? #5673 ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue)tags/v0.17.1
| @@ -15,6 +15,7 @@ | |||
| # | |||
| import re | |||
| import threading | |||
| from collections.abc import Iterable | |||
| from urllib.parse import urljoin | |||
| import requests | |||
| @@ -135,6 +136,8 @@ class DefaultRerank(Base): | |||
| else: | |||
| scores = self._model.compute_score(batch_pairs, max_length=max_length) | |||
| scores = sigmoid(np.array(scores)).tolist() | |||
| if not isinstance(scores, Iterable): | |||
| scores = [scores] | |||
| return scores | |||
| def similarity(self, query: str, texts: list): | |||