test: use psycopg over psycopg2 for sqlalchemy tests (#326)

timescale · Dec 20, 2024 · 8c5baf7 · 8c5baf7
1 parent a4298c3
commit 8c5baf7
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 18 deletions.
diff --git a/projects/pgai/pyproject.toml b/projects/pgai/pyproject.toml
@@ -115,5 +115,4 @@ dev-dependencies = [
     "testcontainers==4.8.1",
     "build==1.2.2.post1",
     "twine==5.1.1",
-    "psycopg2==2.9.10",
 ]
diff --git a/projects/pgai/tests/vectorizer/extensions/conftest.py b/projects/pgai/tests/vectorizer/extensions/conftest.py
@@ -32,7 +32,7 @@ def initialized_engine(
     Returns:
         Engine: Configured SQLAlchemy engine
     """
-    engine = create_engine(postgres_container.get_connection_url())
+    engine = create_engine(postgres_container.get_connection_url(driver="psycopg"))
     with engine.connect() as conn:
         conn.execute(text("CREATE EXTENSION IF NOT EXISTS ai CASCADE;"))
         conn.commit()

diff --git a/projects/pgai/uv.lock b/projects/pgai/uv.lock