Merge pull request #708 from ArnoChenFx/fix-mongo-impl

fix mongo impl
This commit is contained in:
zrguo 2025-02-05 01:55:31 +08:00 committed by GitHub
commit ad2a82e5e4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -2,11 +2,14 @@ import os
from tqdm.asyncio import tqdm as tqdm_async
from dataclasses import dataclass
import pipmaster as pm
import np
import numpy as np
if not pm.is_installed("pymongo"):
pm.install("pymongo")
if not pm.is_installed("motor"):
pm.install("motor")
from pymongo import MongoClient
from motor.motor_asyncio import AsyncIOMotorClient
from typing import Union, List, Tuple