Fix chroma integration failing in cloud (#8057)

* moved the pysqlite3 import above chromadb * fix flake issues * added noqa to fix flake issues * fix flake issues * fix flake issues * fix merge conflicts --------- Co-authored-by: dusvyat <13661123+dusvyat@users.noreply.github.com>
mindsdb · Oct 26, 2023 · b5abd44 · b5abd44
1 parent 3e8a8ba
commit b5abd44
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/mindsdb/integrations/handlers/chromadb_handler/chromadb_handler.py b/mindsdb/integrations/handlers/chromadb_handler/chromadb_handler.py
@@ -1,8 +1,6 @@
-import sys
 from collections import OrderedDict
 from typing import List, Optional
 
-import chromadb
 import pandas as pd
 
 from mindsdb.integrations.libs.const import HANDLER_CONNECTION_ARG_TYPE as ARG_TYPE
@@ -18,8 +16,10 @@
 )
 from mindsdb.utilities import log
 
-__import__('pysqlite3')
-sys.modules['sqlite3'] = sys.modules.pop('pysqlite3')
+import sys
+__import__("pysqlite3")
+sys.modules["sqlite3"] = sys.modules.pop("pysqlite3")
+import chromadb  # noqa: E402
 
 
 class ChromaDBHandler(VectorStoreHandler):