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