Explorar o código

Move import within function to speed up init

jherve hai 1 ano
pai
achega
6f03628f17
Modificáronse 1 ficheiros con 3 adicións e 2 borrados
  1. 3 2
      src/de_quoi_parle_le_monde/snapshot_worker.py

+ 3 - 2
src/de_quoi_parle_le_monde/snapshot_worker.py

@@ -3,7 +3,6 @@ import asyncio
 from attrs import frozen
 import traceback
 from loguru import logger
-from sentence_transformers import SentenceTransformer
 
 from de_quoi_parle_le_monde.http import HttpClient
 from de_quoi_parle_le_monde.internet_archive import (
@@ -129,7 +128,7 @@ async def download_all(
 @frozen
 class EmbeddingsWorker:
     storage: Storage
-    model: SentenceTransformer
+    model: Any
 
     def compute_embeddings_for(self, sentences: dict[int, str]):
         logger.debug(f"Computing embeddings for {len(sentences)} sentences")
@@ -154,6 +153,8 @@ class EmbeddingsWorker:
 
     @staticmethod
     def create(storage, model_path):
+        from sentence_transformers import SentenceTransformer
+
         model = SentenceTransformer(model_path)
         return EmbeddingsWorker(storage, model)