connection => session

This commit is contained in:
Luke Van Seters 2021-07-27 19:02:29 -04:00
parent e38f1384ef
commit f72049a4df

View File

@ -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()