Remove (too large) stress test

rapidsai · Nov 11, 2022 · 84f1cf5 · 84f1cf5
1 parent 8260698
commit 84f1cf5
Showing 1 changed file with 1 addition and 4 deletions.
diff --git 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(