Fixed tests

Author: ankane

tests/test_sqlalchemy.py

@@ -79,12 +79,12 @@ class Item(Base):
     __tablename__ = 'sqlalchemy_orm_item'
 
     id: Mapped[int] = mapped_column(Integer, primary_key=True)
-    embedding: Mapped[Vector] = mapped_column(VECTOR(3))
-    half_embedding: Mapped[HalfVector] = mapped_column(HALFVEC(3))
-    binary_embedding: Mapped[Any] = mapped_column(BIT(3))
-    sparse_embedding: Mapped[SparseVector] = mapped_column(SPARSEVEC(3))
-    embeddings: Mapped[list[Vector]] = mapped_column(ARRAY(VECTOR(3)))
-    half_embeddings: Mapped[list[HalfVector]] = mapped_column(ARRAY(HALFVEC(3)))
+    embedding: Mapped[Vector | None] = mapped_column(VECTOR(3))
+    half_embedding: Mapped[HalfVector | None] = mapped_column(HALFVEC(3))
+    binary_embedding: Mapped[Any | None] = mapped_column(BIT(3))
+    sparse_embedding: Mapped[SparseVector | None] = mapped_column(SPARSEVEC(3))
+    embeddings: Mapped[list[Vector] | None] = mapped_column(ARRAY(VECTOR(3)))
+    half_embeddings: Mapped[list[HalfVector] | None] = mapped_column(ARRAY(HALFVEC(3)))
 
 
 Base.metadata.drop_all(setup_engine)

tests/test_sqlmodel.py

@@ -1,5 +1,6 @@
 from pgvector import HalfVector, SparseVector, Vector
 from pgvector.sqlalchemy import VECTOR, HALFVEC, BIT, SPARSEVEC, avg, sum
+from pydantic import ConfigDict
 import pytest
 from sqlalchemy.exc import StatementError
 from sqlmodel import Field, Index, Session, SQLModel, create_engine, delete, select, text
@@ -11,6 +12,7 @@
 
 class Item(SQLModel, table=True):
     __tablename__ = 'sqlmodel_item'  # type: ignore
+    model_config = ConfigDict(arbitrary_types_allowed=True)  # type: ignore
 
     id: int | None = Field(default=None, primary_key=True)
     embedding: Vector | list[float] | None = Field(default=None, sa_type=VECTOR(3))  # type: ignore