Fixed CI

Author: ankane

pgvector/psycopg/vector.py

@@ -3,13 +3,16 @@
 from psycopg.adapt import Loader, Dumper
 from psycopg.pq import Format
 from psycopg.types import TypeInfo
-from typing import Any, TypeAlias, TYPE_CHECKING
+from typing import Any, TypeAlias
 from .. import Vector
 
 Buffer: TypeAlias = bytes | bytearray | memoryview
 
-if TYPE_CHECKING:
+try:
     import numpy as np
+    NUMPY_AVAILABLE = True
+except ImportError:
+    NUMPY_AVAILABLE = False
 
 
 class VectorDumper(Dumper):
@@ -58,9 +61,11 @@ def register_vector_info(context: BaseConnection[Any], info: TypeInfo | None) ->
     binary_dumper = type('', (VectorBinaryDumper,), {'oid': info.oid})
 
     adapters = context.adapters
-    adapters.register_dumper('numpy.ndarray', text_dumper)
-    adapters.register_dumper('numpy.ndarray', binary_dumper)
     adapters.register_dumper(Vector, text_dumper)
     adapters.register_dumper(Vector, binary_dumper)
     adapters.register_loader(info.oid, VectorLoader)
     adapters.register_loader(info.oid, VectorBinaryLoader)
+
+    if NUMPY_AVAILABLE:
+        adapters.register_dumper(np.ndarray, text_dumper)
+        adapters.register_dumper(np.ndarray, binary_dumper)

pgvector/psycopg2/vector.py

@@ -1,9 +1,11 @@
 from psycopg2.extensions import adapt, connection, cursor, new_array_type, new_type, register_adapter, register_type
-from typing import TYPE_CHECKING
 from .. import Vector
 
-if TYPE_CHECKING:
+try:
     import numpy as np
+    NUMPY_AVAILABLE = True
+except ImportError:
+    NUMPY_AVAILABLE = False
 
 
 class VectorAdapter:
@@ -32,8 +34,5 @@ def register_vector_info(oid: int, array_oid: int | None, scope: connection | cu
 
     register_adapter(Vector, VectorAdapter)
 
-    try:
-        import numpy as np
+    if NUMPY_AVAILABLE:
         register_adapter(np.ndarray, VectorAdapter)
-    except ImportError:
-        pass