Skip to content

Commit 325c214

Browse files
committed
fix sentence transformer import
1 parent ed21570 commit 325c214

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/zotero_arxiv_daily/reranker/local.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,10 +5,10 @@
55
@register_reranker("local")
66
class LocalReranker(BaseReranker):
77
def get_similarity_score(self, s1: list[str], s2: list[str]) -> np.ndarray:
8+
from sentence_transformers import SentenceTransformer
89
if not self.config.executor.debug:
910
from transformers.utils import logging as transformers_logging
1011
from huggingface_hub.utils import logging as hf_logging
11-
from sentence_transformers import SentenceTransformer
1212

1313
transformers_logging.set_verbosity_error()
1414
hf_logging.set_verbosity_error()

0 commit comments

Comments
 (0)