Fixes sqlite version error

NERC-CEH · Nov 8, 2024 · bb843b1 · bb843b1
1 parent 63f9d0c
commit bb843b1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/scripts/upload_to_docstore.py b/scripts/upload_to_docstore.py
@@ -4,6 +4,10 @@
 import uuid
 from argparse import ArgumentParser
 
+__import__("pysqlite3")
+import sys
+
+sys.modules["sqlite3"] = sys.modules.pop("pysqlite3")
 import chromadb
 from chromadb.utils import embedding_functions
 from chromadb.utils.batch_utils import create_batches
@@ -44,7 +48,7 @@ def main(
                 embeddings=batch[1],
                 ids=batch[0],
             )
-            
+
 
 if __name__ == "__main__":
     parser = ArgumentParser("prepare_data.py")