Skip to content

Commit 03d1135

Browse files
committed
Improved type hints [skip ci]
1 parent 3a067c6 commit 03d1135

3 files changed

Lines changed: 6 additions & 9 deletions

File tree

pgvector/psycopg2/halfvec.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,12 @@
11
from psycopg2.extensions import adapt, connection, cursor, new_array_type, new_type, register_adapter, register_type
2-
from typing import Any
32
from .. import HalfVector
43

54

65
class HalfvecAdapter:
7-
def __init__(self, value: Any) -> None:
6+
def __init__(self, value: object) -> None:
87
self._value = value
98

10-
def getquoted(self) -> Any:
9+
def getquoted(self) -> bytes:
1110
return adapt(HalfVector._to_db(self._value)).getquoted()
1211

1312

pgvector/psycopg2/sparsevec.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,12 @@
11
from psycopg2.extensions import adapt, connection, cursor, new_array_type, new_type, register_adapter, register_type
2-
from typing import Any
32
from .. import SparseVector
43

54

65
class SparsevecAdapter:
7-
def __init__(self, value: Any) -> None:
6+
def __init__(self, value: object) -> None:
87
self._value = value
98

10-
def getquoted(self) -> Any:
9+
def getquoted(self) -> bytes:
1110
return adapt(SparseVector._to_db(self._value)).getquoted()
1211

1312

pgvector/psycopg2/vector.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,13 @@
11
import numpy as np
22
from psycopg2.extensions import adapt, connection, cursor, new_array_type, new_type, register_adapter, register_type
3-
from typing import Any
43
from .. import Vector
54

65

76
class VectorAdapter:
8-
def __init__(self, value: Any) -> None:
7+
def __init__(self, value: object) -> None:
98
self._value = value
109

11-
def getquoted(self) -> Any:
10+
def getquoted(self) -> bytes:
1211
return adapt(Vector._to_db(self._value)).getquoted()
1312

1413

0 commit comments

Comments
 (0)