From f72049a4dfceea112bf2b743527954c8a5a4ef60 Mon Sep 17 00:00:00 2001 From: Luke Van Seters Date: Tue, 27 Jul 2021 19:02:29 -0400 Subject: [PATCH] connection => session --- mev_inspect/crud/classified_traces.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mev_inspect/crud/classified_traces.py b/mev_inspect/crud/classified_traces.py index a175d82..15059c5 100644 --- a/mev_inspect/crud/classified_traces.py +++ b/mev_inspect/crud/classified_traces.py @@ -6,12 +6,12 @@ from mev_inspect.schemas.classified_traces import ClassifiedTrace def write_classified_traces( - db_connection, + db_session, classified_traces: List[ClassifiedTrace], ) -> None: models = [ ClassifiedTraceModel(**json.loads(trace.json())) for trace in classified_traces ] - db_connection.bulk_save_objects(models) - db_connection.commit() + db_session.bulk_save_objects(models) + db_session.commit()