From a8db7f56d24792a8f52e3021698cf820e55d595c Mon Sep 17 00:00:00 2001 From: Chang She <759245+changhiskhan@users.noreply.github.com> Date: Tue, 25 Apr 2023 20:08:18 -0700 Subject: [PATCH] tolerance --- python/tests/test_query.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/tests/test_query.py b/python/tests/test_query.py index ae1bebda..9ad7c928 100644 --- a/python/tests/test_query.py +++ b/python/tests/test_query.py @@ -72,7 +72,8 @@ def test_query_builder_with_metric(table): df_cosine = LanceQueryBuilder(table, query).metric("cosine").limit(1).to_df() assert df_cosine.score[0] == pytest.approx( - cosine_distance(query, df_cosine.vector[0]) + cosine_distance(query, df_cosine.vector[0]), + abs=1e-6, ) assert 0 <= df_cosine.score[0] <= 1