diff --git a/backends/infinity/model.py b/backends/infinity/model.py index 35a4df45..c48a42c2 100644 --- 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: