Updated style

Author: ankane

pgvector/psycopg/vector.py

@@ -1,3 +1,4 @@
+from __future__ import annotations
 import psycopg
 from psycopg import BaseConnection
 from psycopg.adapt import Loader, Dumper
@@ -18,7 +19,7 @@
 class VectorDumper(Dumper):
     format = Format.TEXT
 
-    def dump(self, obj: 'Vector | np.ndarray[tuple[int, ...], np.dtype[np.floating]]') -> Buffer | None:
+    def dump(self, obj: Vector | np.ndarray[tuple[int, ...], np.dtype[np.floating]]) -> Buffer | None:
         if not isinstance(obj, Vector):
             obj = Vector(obj)
         return obj.to_text().encode('utf8')
@@ -27,7 +28,7 @@ def dump(self, obj: 'Vector | np.ndarray[tuple[int, ...], np.dtype[np.floating]]
 class VectorBinaryDumper(VectorDumper):
     format = Format.BINARY
 
-    def dump(self, obj: 'Vector | np.ndarray[tuple[int, ...], np.dtype[np.floating]]') -> Buffer | None:
+    def dump(self, obj: Vector | np.ndarray[tuple[int, ...], np.dtype[np.floating]]) -> Buffer | None:
         if not isinstance(obj, Vector):
             obj = Vector(obj)
         return obj.to_binary()

pgvector/psycopg2/vector.py

@@ -1,3 +1,4 @@
+from __future__ import annotations
 from psycopg2.extensions import adapt, connection, cursor, new_array_type, new_type, register_adapter, register_type
 from typing import Any
 from .. import Vector
@@ -10,7 +11,7 @@
 
 
 class VectorAdapter:
-    def __init__(self, value: 'Vector | np.ndarray[tuple[int, ...], np.dtype[np.floating]]') -> None:
+    def __init__(self, value: Vector | np.ndarray[tuple[int, ...], np.dtype[np.floating]]) -> None:
         if not isinstance(value, Vector):
             value = Vector(value)
         self._value = value