From 57989ef8201e806c66b03bd4424fc847632b8afb Mon Sep 17 00:00:00 2001 From: jinfeng Date: Fri, 15 Mar 2024 12:26:26 -0700 Subject: [PATCH] double nrows in test case when nclasses is larger than 5 --- python/cuml/tests/dask/test_dask_logistic_regression.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/cuml/tests/dask/test_dask_logistic_regression.py b/python/cuml/tests/dask/test_dask_logistic_regression.py index b082944ae7..f95020ade2 100644 --- a/python/cuml/tests/dask/test_dask_logistic_regression.py +++ b/python/cuml/tests/dask/test_dask_logistic_regression.py @@ -720,9 +720,9 @@ def test_standardization_on_scaled_dataset( penalty = regularization[0] C = regularization[1] l1_ratio = regularization[2] - nrows = int(1e5) - ncols = ncol_and_nclasses[0] n_classes = ncol_and_nclasses[1] + nrows = int(1e5) if n_classes < 5 else int(2e5) + ncols = ncol_and_nclasses[0] n_info = ncols n_redundant = 0 n_parts = 2