File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -19,7 +19,7 @@ class Item(BaseModel):
1919 binary_embedding = FixedBitField (max_length = 3 , null = True )
2020 sparse_embedding = SparseVectorField (dimensions = 3 , null = True )
2121
22- class Meta :
22+ class Meta : # type: ignore
2323 table_name = 'peewee_item'
2424
2525
@@ -161,7 +161,7 @@ def test_sparsevec_l1_distance(self):
161161
162162 def test_where (self ):
163163 create_items ()
164- items = Item .select ().where (Item .embedding .l2_distance ([1 , 1 , 1 ]) < 1 )
164+ items = Item .select ().where (Item .embedding .l2_distance ([1 , 1 , 1 ]) < 1 ) # type: ignore
165165 assert [v .id for v in items ] == [1 ]
166166
167167 def test_vector_avg (self ):
@@ -207,9 +207,9 @@ def test_vector_array(self):
207207 ext_db = PostgresqlExtDatabase ('pgvector_python_test' )
208208
209209 class ExtItem (BaseModel ):
210- embeddings = ArrayField (VectorField , field_kwargs = {'dimensions' : 3 }, index = False )
210+ embeddings = ArrayField (VectorField , field_kwargs = {'dimensions' : 3 }, index = False ) # type: ignore
211211
212- class Meta :
212+ class Meta : # type: ignore
213213 database = ext_db
214214 table_name = 'peewee_ext_item'
215215
You can’t perform that action at this time.
0 commit comments