Преглед изворни кода

Move import within function to speed up init

jherve пре 1 година
родитељ
комит
6f03628f17
1 измењених фајлова са 3 додато и 2 уклоњено
  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
 from attrs import frozen
 import traceback
 import traceback
 from loguru import logger
 from loguru import logger
-from sentence_transformers import SentenceTransformer
 
 
 from de_quoi_parle_le_monde.http import HttpClient
 from de_quoi_parle_le_monde.http import HttpClient
 from de_quoi_parle_le_monde.internet_archive import (
 from de_quoi_parle_le_monde.internet_archive import (
@@ -129,7 +128,7 @@ async def download_all(
 @frozen
 @frozen
 class EmbeddingsWorker:
 class EmbeddingsWorker:
     storage: Storage
     storage: Storage
-    model: SentenceTransformer
+    model: Any
 
 
     def compute_embeddings_for(self, sentences: dict[int, str]):
     def compute_embeddings_for(self, sentences: dict[int, str]):
         logger.debug(f"Computing embeddings for {len(sentences)} sentences")
         logger.debug(f"Computing embeddings for {len(sentences)} sentences")
@@ -154,6 +153,8 @@ class EmbeddingsWorker:
 
 
     @staticmethod
     @staticmethod
     def create(storage, model_path):
     def create(storage, model_path):
+        from sentence_transformers import SentenceTransformer
+
         model = SentenceTransformer(model_path)
         model = SentenceTransformer(model_path)
         return EmbeddingsWorker(storage, model)
         return EmbeddingsWorker(storage, model)