Skip to content

Commit 057740b

Browse files
committed
Reduced dependency on NumPy [skip ci]
1 parent 2155f99 commit 057740b

3 files changed

Lines changed: 10 additions & 5 deletions

File tree

pgvector/pg8000/register.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
import numpy as np
21
from pg8000.native import Connection
32
from .. import Vector, HalfVector, SparseVector
43

@@ -12,7 +11,11 @@ def register_vector(conn: Connection) -> None:
1211
raise RuntimeError('vector type not found in the database')
1312

1413
conn.register_out_adapter(Vector, Vector._to_db)
15-
conn.register_out_adapter(np.ndarray, Vector._to_db)
14+
try:
15+
import numpy as np
16+
conn.register_out_adapter(np.ndarray, Vector._to_db)
17+
except ImportError:
18+
pass
1619
conn.register_in_adapter(type_info['vector'], Vector._from_db)
1720

1821
if 'halfvec' in type_info:

pgvector/psycopg/vector.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
import numpy as np
21
import psycopg
32
from psycopg import BaseConnection
43
from psycopg.adapt import Loader, Dumper

pgvector/psycopg2/vector.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
import numpy as np
21
from psycopg2.extensions import adapt, connection, cursor, new_array_type, new_type, register_adapter, register_type
32
from .. import Vector
43

@@ -23,5 +22,9 @@ def register_vector_info(oid: int, array_oid: int | None, scope: connection | cu
2322
vectorarray = new_array_type((array_oid,), 'VECTORARRAY', vector)
2423
register_type(vectorarray, scope)
2524

26-
register_adapter(np.ndarray, VectorAdapter)
2725
register_adapter(Vector, VectorAdapter)
26+
try:
27+
import numpy as np
28+
register_adapter(np.ndarray, VectorAdapter)
29+
except ImportError:
30+
pass

0 commit comments

Comments
 (0)