Skip to content

Commit c912f50

Browse files
committed
Simplified _to_db_binary [skip ci]
1 parent 2c95c2a commit c912f50

3 files changed

Lines changed: 3 additions & 12 deletions

File tree

pgvector/halfvec.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -94,10 +94,7 @@ def _to_db(cls, value: object) -> str | None:
9494
return value.to_text()
9595

9696
@classmethod
97-
def _to_db_binary(cls, value: object) -> bytes | None:
98-
if value is None:
99-
return value
100-
97+
def _to_db_binary(cls, value: object) -> bytes:
10198
if not isinstance(value, cls):
10299
value = cls(value) # type: ignore
103100

pgvector/sparsevec.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -159,10 +159,7 @@ def _to_db(cls, value: object) -> str | None:
159159
return value.to_text()
160160

161161
@classmethod
162-
def _to_db_binary(cls, value: object) -> bytes | None:
163-
if value is None:
164-
return value
165-
162+
def _to_db_binary(cls, value: object) -> bytes:
166163
if not isinstance(value, cls):
167164
value = cls(value)
168165

pgvector/vector.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -92,10 +92,7 @@ def _to_db(cls, value: object) -> str | None:
9292
return value.to_text()
9393

9494
@classmethod
95-
def _to_db_binary(cls, value: object) -> bytes | None:
96-
if value is None:
97-
return value
98-
95+
def _to_db_binary(cls, value: object) -> bytes:
9996
if not isinstance(value, cls):
10097
value = cls(value) # type: ignore
10198

0 commit comments

Comments
 (0)