Skip to content

Commit 2281d35

Browse files
modifies test tolerance
1 parent c105447 commit 2281d35

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/integration/test_vectorizers.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import os
2-
32
import numpy as np
3+
44
import pytest
55

66
from redisvl.redis.utils import buffer_to_array
@@ -239,18 +239,18 @@ def bad_return_type(text: str) -> str:
239239
custom_embed_func, embed_many=bad_return_type
240240
)
241241

242-
243242
def test_dtypes(vectorizer):
243+
if isinstance(vectorizer, CustomTextVectorizer):
244+
pytest.skip("skipping custom text vectorizer")
244245
words = "hello"
245-
246246
raw = vectorizer.embed(words, as_buffer=False)
247247

248248
default = vectorizer.embed(words, as_buffer=True)
249249
assert buffer_to_array(default, dtype="float32") == raw
250250

251251
float16 = vectorizer.embed(words, as_buffer=True, dtype="float16")
252252
# assert buffer_to_array(float16, dtype="float16") == raw # fails
253-
assert np.allclose(buffer_to_array(float16, dtype="float16"), raw, atol=1e-04)
253+
assert np.allclose(buffer_to_array(float16, dtype="float16"), raw, atol=1e-03)
254254

255255
float32 = vectorizer.embed(words, as_buffer=True, dtype="float32")
256256
assert buffer_to_array(float32, dtype="float32") == raw

0 commit comments

Comments
 (0)