Switched to import for Buffer

Author: ankane

pgvector/psycopg/bit.py

@@ -1,12 +1,11 @@
 from psycopg import BaseConnection
-from psycopg.types import TypeInfo
+from psycopg.abc import Buffer
 from psycopg.adapt import Dumper
 from psycopg.pq import Format
-from typing import Any, TypeAlias
+from psycopg.types import TypeInfo
+from typing import Any
 from .. import Bit
 
-Buffer: TypeAlias = bytes | bytearray | memoryview
-
 
 class BitDumper(Dumper):
     format = Format.TEXT

pgvector/psycopg/halfvec.py

@@ -1,12 +1,11 @@
 from psycopg import BaseConnection
+from psycopg.abc import Buffer
 from psycopg.adapt import Loader, Dumper
 from psycopg.pq import Format
 from psycopg.types import TypeInfo
-from typing import Any, TypeAlias
+from typing import Any
 from .. import HalfVector
 
-Buffer: TypeAlias = bytes | bytearray | memoryview
-
 
 class HalfVectorDumper(Dumper):
     format = Format.TEXT

pgvector/psycopg/sparsevec.py

@@ -1,12 +1,11 @@
 from psycopg import BaseConnection
+from psycopg.abc import Buffer
 from psycopg.adapt import Loader, Dumper
 from psycopg.pq import Format
 from psycopg.types import TypeInfo
-from typing import Any, TypeAlias
+from typing import Any
 from .. import SparseVector
 
-Buffer: TypeAlias = bytes | bytearray | memoryview
-
 
 class SparseVectorDumper(Dumper):
     format = Format.TEXT

pgvector/psycopg/vector.py

@@ -1,14 +1,13 @@
 from __future__ import annotations
 import psycopg
 from psycopg import BaseConnection
+from psycopg.abc import Buffer
 from psycopg.adapt import Loader, Dumper
 from psycopg.pq import Format
 from psycopg.types import TypeInfo
-from typing import TYPE_CHECKING, Any, TypeAlias
+from typing import TYPE_CHECKING, Any
 from .. import Vector
 
-Buffer: TypeAlias = bytes | bytearray | memoryview
-
 if TYPE_CHECKING:
     from .._utils import ndarray