connection => session
This commit is contained in:
parent
e38f1384ef
commit
f72049a4df
@ -6,12 +6,12 @@ from mev_inspect.schemas.classified_traces import ClassifiedTrace
|
|||||||
|
|
||||||
|
|
||||||
def write_classified_traces(
|
def write_classified_traces(
|
||||||
db_connection,
|
db_session,
|
||||||
classified_traces: List[ClassifiedTrace],
|
classified_traces: List[ClassifiedTrace],
|
||||||
) -> None:
|
) -> None:
|
||||||
models = [
|
models = [
|
||||||
ClassifiedTraceModel(**json.loads(trace.json())) for trace in classified_traces
|
ClassifiedTraceModel(**json.loads(trace.json())) for trace in classified_traces
|
||||||
]
|
]
|
||||||
|
|
||||||
db_connection.bulk_save_objects(models)
|
db_session.bulk_save_objects(models)
|
||||||
db_connection.commit()
|
db_session.commit()
|
||||||
|
Loading…
x
Reference in New Issue
Block a user