From 84f1cf574a23864485bd50d6b8ccdef996640fc4 Mon Sep 17 00:00:00 2001 From: Allard Hendriksen Date: Fri, 11 Nov 2022 10:56:17 +0100 Subject: [PATCH] Remove (too large) stress test --- python/cuml/tests/test_linear_model.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/python/cuml/tests/test_linear_model.py b/python/cuml/tests/test_linear_model.py index 69647d727b..1c8edfd87c 100644 --- a/python/cuml/tests/test_linear_model.py +++ b/python/cuml/tests/test_linear_model.py @@ -88,10 +88,7 @@ def make_classification_dataset(datatype, nrows, ncols, n_info, num_classes): return X_train, X_test, y_train, y_test -@pytest.mark.parametrize( - "ntargets", - [unit_param(1), unit_param(2), quality_param(100), stress_param(1000)] -) +@pytest.mark.parametrize("ntargets", [1, 2]) @pytest.mark.parametrize("datatype", [np.float32, np.float64]) @pytest.mark.parametrize("algorithm", ["eig", "svd"]) @pytest.mark.parametrize(