Skip to content

Commit 56283d4

Browse files
committed
Fixed typechecking [skip ci]
1 parent 087a2df commit 56283d4

6 files changed

Lines changed: 9 additions & 6 deletions

File tree

‎pgvector/django/halfvec.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ def formfield(self, form_class: Any = None, choices_form_class: Any = None, **kw
5151

5252

5353
class HalfVectorWidget(forms.TextInput):
54-
def format_value(self, value: Any) -> str | None:
54+
def format_value(self, value: Any) -> Any:
5555
if isinstance(value, HalfVector):
5656
value = value.to_list()
5757
return super().format_value(value)

‎pgvector/django/sparsevec.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ def formfield(self, form_class: Any = None, choices_form_class: Any = None, **kw
4646

4747

4848
class SparseVectorWidget(forms.TextInput):
49-
def format_value(self, value: Any) -> str | None:
49+
def format_value(self, value: Any) -> Any:
5050
if isinstance(value, SparseVector):
5151
value = value.to_text()
5252
return super().format_value(value)

‎pgvector/django/vector.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ def formfield(self, form_class: Any = None, choices_form_class: Any = None, **kw
5151

5252

5353
class VectorWidget(forms.TextInput):
54-
def format_value(self, value: Any) -> str | None:
54+
def format_value(self, value: Any) -> Any:
5555
if isinstance(value, Vector):
5656
value = value.to_list()
5757
return super().format_value(value)

‎pgvector/psycopg2/halfvec.py‎

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

45

56
class HalfvecAdapter:
67
def __init__(self, value: HalfVector) -> None:
78
self._value = value
89

9-
def getquoted(self) -> bytes:
10+
def getquoted(self) -> Any:
1011
return adapt(self._value.to_text()).getquoted()
1112

1213

‎pgvector/psycopg2/sparsevec.py‎

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

45

56
class SparsevecAdapter:
67
def __init__(self, value: SparseVector) -> None:
78
self._value = value
89

9-
def getquoted(self) -> bytes:
10+
def getquoted(self) -> Any:
1011
return adapt(self._value.to_text()).getquoted()
1112

1213

‎pgvector/psycopg2/vector.py‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
from psycopg2.extensions import adapt, connection, cursor, new_array_type, new_type, register_adapter, register_type
2+
from typing import Any
23
from .. import Vector
34

45
try:
@@ -14,7 +15,7 @@ def __init__(self, value: 'Vector | np.ndarray') -> None:
1415
value = Vector(value)
1516
self._value = value
1617

17-
def getquoted(self) -> bytes:
18+
def getquoted(self) -> Any:
1819
return adapt(self._value.to_text()).getquoted()
1920

2021

0 commit comments

Comments
 (0)