瀏覽代碼

Move import back to start of file

jherve 1 年之前
父節點
當前提交
1599204563
共有 1 個文件被更改,包括 1 次插入2 次删除
  1. 1 2
      src/de_quoi_parle_le_monde/embeddings.py

+ 1 - 2
src/de_quoi_parle_le_monde/embeddings.py

@@ -5,6 +5,7 @@ from collections import defaultdict
 from typing import Any
 from attrs import frozen
 from numpy.typing import NDArray
+from sentence_transformers import SentenceTransformer
 
 from de_quoi_parle_le_monde.storage import Storage
 
@@ -78,8 +79,6 @@ class EmbeddingsWorker:
 
     @staticmethod
     def create(storage, model_path):
-        from sentence_transformers import SentenceTransformer
-
         model = SentenceTransformer(model_path)
         return EmbeddingsWorker(storage, model)