Skip to content

Commit 2bf00f8

Browse files
committed
Updated style [skip ci]
1 parent 77d878b commit 2bf00f8

2 files changed

Lines changed: 3 additions & 1 deletion

File tree

pgvector/pg8000/register.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,12 +11,13 @@ def register_vector(conn: Connection) -> None:
1111
raise RuntimeError('vector type not found in the database')
1212

1313
conn.register_out_adapter(Vector, Vector._to_db)
14+
conn.register_in_adapter(type_info['vector'], Vector._from_db)
15+
1416
try:
1517
import numpy as np
1618
conn.register_out_adapter(np.ndarray, Vector._to_db)
1719
except ImportError:
1820
pass
19-
conn.register_in_adapter(type_info['vector'], Vector._from_db)
2021

2122
if 'halfvec' in type_info:
2223
conn.register_out_adapter(HalfVector, HalfVector._to_db)

pgvector/psycopg2/vector.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@ def register_vector_info(oid: int, array_oid: int | None, scope: connection | cu
2323
register_type(vectorarray, scope)
2424

2525
register_adapter(Vector, VectorAdapter)
26+
2627
try:
2728
import numpy as np
2829
register_adapter(np.ndarray, VectorAdapter)

0 commit comments

Comments
 (0)