Updated examples [skip ci]

Author: ankane

examples/implicit/example.py

@@ -1,6 +1,6 @@
 import implicit
 from implicit.datasets.movielens import get_movielens
-from pgvector.sqlalchemy import Vector
+from pgvector.sqlalchemy import VECTOR
 from sqlalchemy import create_engine, insert, select, text, Integer, String
 from sqlalchemy.orm import declarative_base, mapped_column, Session
 
@@ -16,15 +16,15 @@ class User(Base):
     __tablename__ = 'user'
 
     id = mapped_column(Integer, primary_key=True)
-    factors = mapped_column(Vector(20))
+    factors = mapped_column(VECTOR(20))
 
 
 class Item(Base):
     __tablename__ = 'item'
 
     id = mapped_column(Integer, primary_key=True)
     title = mapped_column(String)
-    factors = mapped_column(Vector(20))
+    factors = mapped_column(VECTOR(20))
 
 
 Base.metadata.drop_all(engine)

examples/lightfm/example.py

@@ -1,6 +1,6 @@
 from lightfm import LightFM
 from lightfm.datasets import fetch_movielens
-from pgvector.sqlalchemy import Vector
+from pgvector.sqlalchemy import VECTOR
 from sqlalchemy import create_engine, insert, select, text, Float, Integer, String
 from sqlalchemy.orm import declarative_base, mapped_column, Session
 
@@ -16,15 +16,15 @@ class User(Base):
     __tablename__ = 'user'
 
     id = mapped_column(Integer, primary_key=True)
-    factors = mapped_column(Vector(20))
+    factors = mapped_column(VECTOR(20))
 
 
 class Item(Base):
     __tablename__ = 'item'
 
     id = mapped_column(Integer, primary_key=True)
     title = mapped_column(String)
-    factors = mapped_column(Vector(20))
+    factors = mapped_column(VECTOR(20))
     bias = mapped_column(Float)
 
 

examples/surprise/example.py

@@ -1,4 +1,4 @@
-from pgvector.sqlalchemy import Vector
+from pgvector.sqlalchemy import VECTOR
 from sqlalchemy import create_engine, insert, select, text, Integer
 from sqlalchemy.orm import declarative_base, mapped_column, Session
 from surprise import Dataset, SVD
@@ -15,14 +15,14 @@ class User(Base):
     __tablename__ = 'user'
 
     id = mapped_column(Integer, primary_key=True)
-    factors = mapped_column(Vector(20))
+    factors = mapped_column(VECTOR(20))
 
 
 class Item(Base):
     __tablename__ = 'item'
 
     id = mapped_column(Integer, primary_key=True)
-    factors = mapped_column(Vector(20))
+    factors = mapped_column(VECTOR(20))
 
 
 Base.metadata.drop_all(engine)