Merge pull request theroyallab#184 from SecretiveShell/Infinity-Embed…

…-TODO Complete conditional infinity import TODO
SecretiveShell · Sep 5, 2024 · 8524999 · 8524999
2 parents 03ff472 + 42a42ca
commit 8524999
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/backends/infinity/model.py b/backends/infinity/model.py
@@ -7,13 +7,13 @@
 from common.utils import unwrap
 
 # Conditionally import infinity to sidestep its logger
-# TODO: Make this prettier
+has_infinity_emb: bool = False
 try:
     from infinity_emb import EngineArgs, AsyncEmbeddingEngine
 
     has_infinity_emb = True
 except ImportError:
-    has_infinity_emb = False
+    pass
 
 
 class InfinityContainer: