Skip to content

Commit fb571b5

Browse files
committed
Updated get_or_create test
1 parent 01159f5 commit fb571b5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_peewee.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from math import sqrt
22
import numpy as np
3-
from peewee import AsIs, Model, PostgresqlDatabase
3+
from peewee import Model, PostgresqlDatabase
44
from pgvector.peewee import VectorField
55

66
db = PostgresqlDatabase('pgvector_python_test')
@@ -72,4 +72,4 @@ def test_where(self):
7272
def test_get_or_create(self):
7373
Item.get_or_create(id=1, defaults={'embedding': [1, 2, 3]})
7474
Item.get_or_create(embedding=np.array([4, 5, 6]))
75-
Item.get_or_create(embedding=AsIs([7, 8, 9]))
75+
Item.get_or_create(embedding=Item.embedding.to_value([7, 8, 9]))

0 commit comments

Comments
 (0)