From dd845cab9a5c5395675bc47ff07ebf3d614fd3bc Mon Sep 17 00:00:00 2001 From: Jim Crist-Harif Date: Thu, 14 May 2026 14:18:10 -0500 Subject: [PATCH 1/2] Mark upstream umap test as flaky This fails sometimes with a trustworthiness score _just_ below 0.85 (e.g. 0.844). --- python/cuml/cuml_accel_tests/upstream/umap/xfail-list.yaml | 1 + 1 file changed, 1 insertion(+) diff --git a/python/cuml/cuml_accel_tests/upstream/umap/xfail-list.yaml b/python/cuml/cuml_accel_tests/upstream/umap/xfail-list.yaml index f4e8d267dd..170b8fefb6 100644 --- a/python/cuml/cuml_accel_tests/upstream/umap/xfail-list.yaml +++ b/python/cuml/cuml_accel_tests/upstream/umap/xfail-list.yaml @@ -37,6 +37,7 @@ marker: cuml_accel_flaky strict: false tests: + - "umap.tests.test_umap_on_iris::test_precomputed_transform_on_iris" - "umap.tests.test_umap_ops::test_umap_transform_embedding_stability" - reason: UMAP using removed numpy function `in1d` marker: cuml_accel_no_numpy_in1d From 81205975fd7258ce5c792e743ea80d8c074ea613 Mon Sep 17 00:00:00 2001 From: Jim Crist-Harif Date: Fri, 15 May 2026 07:48:13 -0500 Subject: [PATCH 2/2] Loosen tolerance on `test_nearest_neighbors_pickle` --- python/cuml/tests/test_pickle.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/cuml/tests/test_pickle.py b/python/cuml/tests/test_pickle.py index e6f9099285..6c9537f64a 100644 --- a/python/cuml/tests/test_pickle.py +++ b/python/cuml/tests/test_pickle.py @@ -473,7 +473,7 @@ def test_nearest_neighbors_pickle(algorithm): # just to ensure things are wired together properly. accuracy = (i1 == i2).sum() / i1.size assert accuracy >= 0.9 - np.testing.assert_allclose(d1, d2, atol=1e-5) + np.testing.assert_allclose(d1, d2, atol=1e-3) else: np.testing.assert_allclose(i1, i2) np.testing.assert_allclose(d1, d2)