Updated examples to use Psycopg 3 [skip ci]

ankane · ankane · commit db8743d2efe1 · 2023-03-31T23:25:15.000-07:00
diff --git a/examples/implicit_recs.py b/examples/implicit_recs.py
@@ -4,7 +4,7 @@
 from sqlalchemy import create_engine, select, text, Integer, String
 from sqlalchemy.orm import declarative_base, mapped_column, Session
 
-engine = create_engine('postgresql+psycopg2://localhost/pgvector_example')
+engine = create_engine('postgresql+psycopg://localhost/pgvector_example')
 with engine.connect() as conn:
     conn.execute(text('CREATE EXTENSION IF NOT EXISTS vector'))
     conn.commit()
diff --git a/examples/lightfm_recs.py b/examples/lightfm_recs.py
@@ -4,7 +4,7 @@
 from sqlalchemy import create_engine, select, text, Float, Integer, String
 from sqlalchemy.orm import declarative_base, mapped_column, Session
 
-engine = create_engine('postgresql+psycopg2://localhost/pgvector_example')
+engine = create_engine('postgresql+psycopg://localhost/pgvector_example')
 with engine.connect() as conn:
     conn.execute(text('CREATE EXTENSION IF NOT EXISTS vector'))
     conn.commit()
diff --git a/examples/surprise_recs.py b/examples/surprise_recs.py
@@ -3,7 +3,7 @@
 from sqlalchemy.orm import declarative_base, mapped_column, Session
 from surprise import Dataset, SVD
 
-engine = create_engine('postgresql+psycopg2://localhost/pgvector_example')
+engine = create_engine('postgresql+psycopg://localhost/pgvector_example')
 with engine.connect() as conn:
     conn.execute(text('CREATE EXTENSION IF NOT EXISTS vector'))
     conn.commit()
