@@ -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