sqlalchemyy 2 fix

ploomber · Jun 29, 2023 · a68b10a · a68b10a
1 parent 879afd5
commit a68b10a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/sql/run.py b/src/sql/run.py
@@ -652,7 +652,11 @@ def run(conn, sql, config):
                         resultset = ResultSet(result, config, statement, conn.engine)
                         session.commit()
             else:
-                result = conn.engine.execute(statement)
+                # we need the session so sqlalchemy 2.x works
+                # result = conn.engine.execute(statement)
+                with Session(conn.engine) as session:
+                    result = session.execute(statement)
+
                 resultset = ResultSet(result, config, statement, conn.engine)
 
             if result and config.feedback: