Visitar URL original
Improved validation for bit strings - fixes #161 · pgvector/pgvector-python@474522b · GitHub
Skip to content

Commit 474522b

Browse files
committed
Improved validation for bit strings - fixes #161
1 parent 9005637 commit 474522b

1 file changed

Lines changed: 3 additions & 4 deletions

File tree

‎pgvector/bit.py‎

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -26,16 +26,15 @@ def __init__(
2626
value = ''.join([bits[v] for v in value])
2727
except (KeyError, TypeError):
2828
raise ValueError('expected list[bool]')
29+
elif not set(value).issubset({'0', '1'}):
30+
raise ValueError('expected bit string')
2931

3032
length = len(value)
3133
if length % 8 != 0:
3234
value += '0' * (8 - (length % 8))
3335

3436
self._length = length
35-
try:
36-
self._data = int(value, 2).to_bytes(len(value) // 8, byteorder='big')
37-
except ValueError:
38-
raise ValueError('expected bit string')
37+
self._data = int(value, 2).to_bytes(len(value) // 8, byteorder='big')
3938
elif is_ndarray(value):
4039
import numpy as np
4140

0 commit comments

Comments
 (0)