File size: 661 Bytes
5a95d9a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | import datetime
from bson import ObjectId, DatetimeMS
class TestCreator:
def __init__(self, db):
self.db = db
self.test_obj = None
def create(self, record_ids: list[str]) -> ObjectId:
record_results = {}
for batch_id in record_ids:
record_results[batch_id] = {"status": "pending"}
test_obj = {
"status": "pending",
"created_at": datetime.datetime.now(datetime.UTC),
"pipeline_version": "1.0.0",
"record_results": record_results
}
insert_result = self.db.test_evaluation.insert_one(test_obj)
return insert_result.inserted_id
|