There was an error while loading. Please reload this page.
1 parent 45b054c commit a810f27Copy full SHA for a810f27
1 file changed
examples/citus/example.py
@@ -1,4 +1,5 @@
1
import numpy as np
2
+from pgvector import Vector
3
from pgvector.psycopg import register_vector
4
import psycopg
5
@@ -38,7 +39,7 @@
38
39
copy.set_types(['vector', 'bigint'])
40
41
for i in range(rows):
- copy.write_row([embeddings[i], categories[i]])
42
+ copy.write_row([Vector(embeddings[i]), categories[i]])
43
44
print('Creating index in parallel')
45
conn.execute('CREATE INDEX ON items USING hnsw (embedding vector_l2_ops)')
0 commit comments