From d508258f147360ea08c57146a9814fe66f5eac6c Mon Sep 17 00:00:00 2001 From: eabodrov Date: Mon, 25 Dec 2023 09:40:37 +0300 Subject: [PATCH] Fix cql test --- tests/experimental/models/test_cql.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/experimental/models/test_cql.py b/tests/experimental/models/test_cql.py index 93f06deb6..d681f9be1 100644 --- a/tests/experimental/models/test_cql.py +++ b/tests/experimental/models/test_cql.py @@ -138,6 +138,7 @@ def test_predict_pairs_warm_items_only(log, log_to_pred): assert np.allclose( recs_joined.select("relevance").toPandas().to_numpy(), recs_joined.select("pairs_relevance").toPandas().to_numpy(), + atol=0.01 )