Use session [skip ci]

ankane · ankane · commit bd36d645c8b8 · 2023-10-01T21:02:19.000-07:00
diff --git a/tests/test_sqlalchemy.py b/tests/test_sqlalchemy.py
@@ -7,9 +7,9 @@
 from sqlalchemy.sql import func
 
 engine = create_engine('postgresql+psycopg2://localhost/pgvector_python_test')
-with engine.connect() as con:
-    con.execute(text('CREATE EXTENSION IF NOT EXISTS vector'))
-    con.commit()
+with Session(engine) as session:
+    session.execute(text('CREATE EXTENSION IF NOT EXISTS vector'))
+    session.commit()
 
 Base = declarative_base()
 
