Skip to content

Commit 6ef2484

Browse files
committed
Improved type hints [skip ci]
1 parent c636fbf commit 6ef2484

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

tests/test_sqlalchemy.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -535,7 +535,7 @@ def test_binary_quantize(self, engine):
535535
session.add(Item(id=3, embedding=[1, 2, 3]))
536536
session.commit()
537537

538-
distance = func.cast(func.binary_quantize(Item.embedding), BIT(3)).hamming_distance(func.binary_quantize(func.cast([3, -1, 2], VECTOR(3))))
538+
distance = func.cast(func.binary_quantize(Item.embedding), BIT(3)).hamming_distance(func.binary_quantize(func.cast([3, -1, 2], VECTOR(3)))) # type: ignore
539539
items = session.query(Item).order_by(distance).all()
540540
assert [v.id for v in items] == [2, 3, 1]
541541

@@ -550,7 +550,7 @@ def test_binary_quantize_reranking(self, engine):
550550
session.add(Item(id=3, embedding=[1, 2, 3]))
551551
session.commit()
552552

553-
distance = func.cast(func.binary_quantize(Item.embedding), BIT(3)).hamming_distance(func.binary_quantize(func.cast([3, -1, 2], VECTOR(3))))
553+
distance = func.cast(func.binary_quantize(Item.embedding), BIT(3)).hamming_distance(func.binary_quantize(func.cast([3, -1, 2], VECTOR(3)))) # type: ignore
554554
subquery = session.query(Item).order_by(distance).limit(20).subquery()
555555
items = session.query(subquery).order_by(subquery.c.embedding.cosine_distance([3, -1, 2])).limit(5).all()
556556
assert [v.id for v in items] == [2, 3, 1]

0 commit comments

Comments
 (0)