Merge branch 'hotfix/python_310_cql_tests' into 'main'

Fix cql test See merge request ai-lab-pmo/mltools/recsys/RePlay!159
sb-ai-lab · Dec 25, 2023 · 01ce47e · 01ce47e
2 parents e7fd708 + d508258
commit 01ce47e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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
     )