diff --git a/tests/functional/examples/test_notebooks_python.py b/tests/functional/examples/test_notebooks_python.py
index da476cd75f..f7f6134f55 100644
--- a/tests/functional/examples/test_notebooks_python.py
+++ b/tests/functional/examples/test_notebooks_python.py
@@ -17,7 +17,7 @@
         (
             "1m",
             {
-                "map": 0.060579,
+                "map": 0.185099,
                 "ndcg": 0.299245,
                 "precision": 0.270116,
                 "recall": 0.104350,
@@ -26,7 +26,7 @@
         (
             "10m",
             {
-                "map": 0.098745,
+                "map": 0.203722,
                 "ndcg": 0.319625,
                 "precision": 0.275756,
                 "recall": 0.154014,
@@ -57,7 +57,7 @@ def test_sar_single_node_functional(
         (
             "1m",
             {
-                "map": 0.033914,
+                "map": 0.139016,
                 "ndcg": 0.231570,
                 "precision": 0.211923,
                 "recall": 0.064663,
diff --git a/tests/smoke/examples/test_notebooks_gpu.py b/tests/smoke/examples/test_notebooks_gpu.py
index 34e3ce272b..9090b81a91 100644
--- a/tests/smoke/examples/test_notebooks_gpu.py
+++ b/tests/smoke/examples/test_notebooks_gpu.py
@@ -210,7 +210,7 @@ def test_cornac_bivae_smoke(notebooks, output_notebook, kernel_name):
     )
     results = read_notebook(output_notebook)
 
-    assert results["map"] == pytest.approx(0.146552, rel=TOL, abs=ABS_TOL)
+    assert results["map"] == pytest.approx(0.330255, rel=TOL, abs=ABS_TOL)
     assert results["ndcg"] == pytest.approx(0.474124, rel=TOL, abs=ABS_TOL)
     assert results["precision"] == pytest.approx(0.412527, rel=TOL, abs=ABS_TOL)
     assert results["recall"] == pytest.approx(0.225064, rel=TOL, abs=ABS_TOL)
diff --git a/tests/smoke/examples/test_notebooks_python.py b/tests/smoke/examples/test_notebooks_python.py
index c767920e40..04cac4a0e4 100644
--- a/tests/smoke/examples/test_notebooks_python.py
+++ b/tests/smoke/examples/test_notebooks_python.py
@@ -22,7 +22,7 @@ def test_sar_single_node_smoke(notebooks, output_notebook, kernel_name):
     )
     results = read_notebook(output_notebook)
 
-    assert results["map"] == pytest.approx(0.110591, rel=TOL, abs=ABS_TOL)
+    assert results["map"] == pytest.approx(0.244722, rel=TOL, abs=ABS_TOL)
     assert results["ndcg"] == pytest.approx(0.382461, rel=TOL, abs=ABS_TOL)
     assert results["precision"] == pytest.approx(0.330753, rel=TOL, abs=ABS_TOL)
     assert results["recall"] == pytest.approx(0.176385, rel=TOL, abs=ABS_TOL)
@@ -43,7 +43,7 @@ def test_baseline_deep_dive_smoke(notebooks, output_notebook, kernel_name):
     assert results["mae"] == pytest.approx(0.846033, rel=TOL, abs=ABS_TOL)
     assert results["rsquared"] == pytest.approx(0.136435, rel=TOL, abs=ABS_TOL)
     assert results["exp_var"] == pytest.approx(0.136446, rel=TOL, abs=ABS_TOL)
-    assert results["map"] == pytest.approx(0.052850, rel=TOL, abs=ABS_TOL)
+    assert results["map"] == pytest.approx(0.138672, rel=TOL, abs=ABS_TOL)
     assert results["ndcg"] == pytest.approx(0.248061, rel=TOL, abs=ABS_TOL)
     assert results["precision"] == pytest.approx(0.223754, rel=TOL, abs=ABS_TOL)
     assert results["recall"] == pytest.approx(0.108826, rel=TOL, abs=ABS_TOL)