Skip to content

Commit 8598dc0

Browse files
ankaneankane
committed
Improved type hints [skip ci]
1 parent b4b684d commit 8598dc0

8 files changed

Lines changed: 32 additions & 32 deletions

File tree

‎pgvector/peewee/bit.py‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,11 @@ def __init__(self, max_length: int | None = None, *args: Any, **kwargs: Any) ->
1212
def get_modifiers(self) -> list[int] | None:
1313
return [self.max_length] if self.max_length else None
1414

15-
def _distance(self, op: str, vector: Any) -> Expression:
15+
def _distance(self, op: str, vector: object) -> Expression:
1616
return Expression(lhs=self, op=op, rhs=self.to_value(vector))
1717

18-
def hamming_distance(self, vector: Any) -> Expression:
18+
def hamming_distance(self, vector: object) -> Expression:
1919
return self._distance('<~>', vector)
2020

21-
def jaccard_distance(self, vector: Any) -> Expression:
21+
def jaccard_distance(self, vector: object) -> Expression:
2222
return self._distance('<%%>', vector)

‎pgvector/peewee/halfvec.py‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -19,17 +19,17 @@ def db_value(self, value: Any) -> str | None:
1919
def python_value(self, value: Any) -> HalfVector | None:
2020
return HalfVector._from_db(value)
2121

22-
def _distance(self, op: str, vector: Any) -> Expression:
22+
def _distance(self, op: str, vector: object) -> Expression:
2323
return Expression(lhs=self, op=op, rhs=self.to_value(vector))
2424

25-
def l2_distance(self, vector: Any) -> Expression:
25+
def l2_distance(self, vector: object) -> Expression:
2626
return self._distance('<->', vector)
2727

28-
def max_inner_product(self, vector: Any) -> Expression:
28+
def max_inner_product(self, vector: object) -> Expression:
2929
return self._distance('<#>', vector)
3030

31-
def cosine_distance(self, vector: Any) -> Expression:
31+
def cosine_distance(self, vector: object) -> Expression:
3232
return self._distance('<=>', vector)
3333

34-
def l1_distance(self, vector: Any) -> Expression:
34+
def l1_distance(self, vector: object) -> Expression:
3535
return self._distance('<+>', vector)

‎pgvector/peewee/sparsevec.py‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -19,17 +19,17 @@ def db_value(self, value: Any) -> str | None:
1919
def python_value(self, value: Any) -> SparseVector | None:
2020
return SparseVector._from_db(value)
2121

22-
def _distance(self, op: str, vector: Any) -> Expression:
22+
def _distance(self, op: str, vector: object) -> Expression:
2323
return Expression(lhs=self, op=op, rhs=self.to_value(vector))
2424

25-
def l2_distance(self, vector: Any) -> Expression:
25+
def l2_distance(self, vector: object) -> Expression:
2626
return self._distance('<->', vector)
2727

28-
def max_inner_product(self, vector: Any) -> Expression:
28+
def max_inner_product(self, vector: object) -> Expression:
2929
return self._distance('<#>', vector)
3030

31-
def cosine_distance(self, vector: Any) -> Expression:
31+
def cosine_distance(self, vector: object) -> Expression:
3232
return self._distance('<=>', vector)
3333

34-
def l1_distance(self, vector: Any) -> Expression:
34+
def l1_distance(self, vector: object) -> Expression:
3535
return self._distance('<+>', vector)

‎pgvector/peewee/vector.py‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -20,17 +20,17 @@ def db_value(self, value: Any) -> str | None:
2020
def python_value(self, value: Any) -> np.ndarray | None:
2121
return Vector._from_db(value)
2222

23-
def _distance(self, op: str, vector: Any) -> Expression:
23+
def _distance(self, op: str, vector: object) -> Expression:
2424
return Expression(lhs=self, op=op, rhs=self.to_value(vector))
2525

26-
def l2_distance(self, vector: Any) -> Expression:
26+
def l2_distance(self, vector: object) -> Expression:
2727
return self._distance('<->', vector)
2828

29-
def max_inner_product(self, vector: Any) -> Expression:
29+
def max_inner_product(self, vector: object) -> Expression:
3030
return self._distance('<#>', vector)
3131

32-
def cosine_distance(self, vector: Any) -> Expression:
32+
def cosine_distance(self, vector: object) -> Expression:
3333
return self._distance('<=>', vector)
3434

35-
def l1_distance(self, vector: Any) -> Expression:
35+
def l1_distance(self, vector: object) -> Expression:
3636
return self._distance('<+>', vector)

‎pgvector/sqlalchemy/bit.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,10 +29,10 @@ def process(value: Any) -> Any:
2929
return super().bind_processor(dialect)
3030

3131
class comparator_factory(UserDefinedType.Comparator):
32-
def hamming_distance(self, other: Any) -> Operators:
32+
def hamming_distance(self, other: object) -> Operators:
3333
return self.op('<~>', return_type=Float)(other)
3434

35-
def jaccard_distance(self, other: Any) -> Operators:
35+
def jaccard_distance(self, other: object) -> Operators:
3636
return self.op('<%>', return_type=Float)(other)
3737

3838

‎pgvector/sqlalchemy/halfvec.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -36,16 +36,16 @@ def process(value: Any) -> HalfVector | None:
3636
return process
3737

3838
class comparator_factory(UserDefinedType.Comparator):
39-
def l2_distance(self, other: Any) -> Operators:
39+
def l2_distance(self, other: object) -> Operators:
4040
return self.op('<->', return_type=Float)(other)
4141

42-
def max_inner_product(self, other: Any) -> Operators:
42+
def max_inner_product(self, other: object) -> Operators:
4343
return self.op('<#>', return_type=Float)(other)
4444

45-
def cosine_distance(self, other: Any) -> Operators:
45+
def cosine_distance(self, other: object) -> Operators:
4646
return self.op('<=>', return_type=Float)(other)
4747

48-
def l1_distance(self, other: Any) -> Operators:
48+
def l1_distance(self, other: object) -> Operators:
4949
return self.op('<+>', return_type=Float)(other)
5050

5151

‎pgvector/sqlalchemy/sparsevec.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -36,16 +36,16 @@ def process(value: Any) -> SparseVector | None:
3636
return process
3737

3838
class comparator_factory(UserDefinedType.Comparator):
39-
def l2_distance(self, other: Any) -> Operators:
39+
def l2_distance(self, other: object) -> Operators:
4040
return self.op('<->', return_type=Float)(other)
4141

42-
def max_inner_product(self, other: Any) -> Operators:
42+
def max_inner_product(self, other: object) -> Operators:
4343
return self.op('<#>', return_type=Float)(other)
4444

45-
def cosine_distance(self, other: Any) -> Operators:
45+
def cosine_distance(self, other: object) -> Operators:
4646
return self.op('<=>', return_type=Float)(other)
4747

48-
def l1_distance(self, other: Any) -> Operators:
48+
def l1_distance(self, other: object) -> Operators:
4949
return self.op('<+>', return_type=Float)(other)
5050

5151

‎pgvector/sqlalchemy/vector.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -37,16 +37,16 @@ def process(value: Any) -> np.ndarray | None:
3737
return process
3838

3939
class comparator_factory(UserDefinedType.Comparator):
40-
def l2_distance(self, other: Any) -> Operators:
40+
def l2_distance(self, other: object) -> Operators:
4141
return self.op('<->', return_type=Float)(other)
4242

43-
def max_inner_product(self, other: Any) -> Operators:
43+
def max_inner_product(self, other: object) -> Operators:
4444
return self.op('<#>', return_type=Float)(other)
4545

46-
def cosine_distance(self, other: Any) -> Operators:
46+
def cosine_distance(self, other: object) -> Operators:
4747
return self.op('<=>', return_type=Float)(other)
4848

49-
def l1_distance(self, other: Any) -> Operators:
49+
def l1_distance(self, other: object) -> Operators:
5050
return self.op('<+>', return_type=Float)(other)
5151

5252

0 commit comments

Comments
 (0)