From f40b7f97ec07d64ffe655c4090606d931020d7db Mon Sep 17 00:00:00 2001 From: viclafargue Date: Tue, 11 Jul 2023 14:16:59 +0200 Subject: [PATCH 1/2] Fix PCA test --- python/cuml/tests/test_pca.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/python/cuml/tests/test_pca.py b/python/cuml/tests/test_pca.py index 9623293c04..65d0e88ab3 100644 --- a/python/cuml/tests/test_pca.py +++ b/python/cuml/tests/test_pca.py @@ -257,13 +257,8 @@ def test_pca_inverse_transform(datatype, input_type, name, use_handle, nrows): cupca.handle.sync() assert array_equal(input_gdf, X, 5e-5, with_sign=True) - -@pytest.mark.skipif( - cp.cuda.driver.get_build_version() <= 11020, - reason="Test failing on driver 11.2", -) -@pytest.mark.parametrize("nrows", [4000, 8000]) -@pytest.mark.parametrize("ncols", [5000, stress_param(20000)]) +@pytest.mark.parametrize("nrows", [4000, 7000]) +@pytest.mark.parametrize("ncols", [2500, stress_param(20000)]) @pytest.mark.parametrize("whiten", [True, False]) @pytest.mark.parametrize("return_sparse", [True, False]) @pytest.mark.parametrize("cupy_input", [True, False]) From 7c4b6b169c6f3bf639f50d2bcc41a69b990434a1 Mon Sep 17 00:00:00 2001 From: Simon Adorf Date: Wed, 12 Jul 2023 08:21:11 -0700 Subject: [PATCH 2/2] Address linter issues. --- python/cuml/tests/test_pca.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/cuml/tests/test_pca.py b/python/cuml/tests/test_pca.py index 65d0e88ab3..10db9a4f7b 100644 --- a/python/cuml/tests/test_pca.py +++ b/python/cuml/tests/test_pca.py @@ -257,6 +257,7 @@ def test_pca_inverse_transform(datatype, input_type, name, use_handle, nrows): cupca.handle.sync() assert array_equal(input_gdf, X, 5e-5, with_sign=True) + @pytest.mark.parametrize("nrows", [4000, 7000]) @pytest.mark.parametrize("ncols", [2500, stress_param(20000)]) @pytest.mark.parametrize("whiten", [True, False])