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