tolerance

This commit is contained in:
Chang She
2023-04-25 20:08:18 -07:00
parent 7a375185a1
commit a8db7f56d2

View File

@@ -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