diff --git a/sklearn/utils/estimator_checks.py b/sklearn/utils/estimator_checks.py index 2108d33d6ad77..d1fe1d5ab8c56 100644 --- a/sklearn/utils/estimator_checks.py +++ b/sklearn/utils/estimator_checks.py @@ -1028,9 +1028,9 @@ def check_array_api_input_and_values( def _check_estimator_sparse_container(name, estimator_orig, sparse_type): rng = np.random.RandomState(0) X = rng.uniform(size=(40, 3)) - X[X < 0.8] = 0 + X[X < 0.6] = 0 X = _enforce_estimator_tags_X(estimator_orig, X) - y = (4 * rng.uniform(size=40)).astype(int) + y = (4 * rng.uniform(size=X.shape[0])).astype(np.int32) # catch deprecation warnings with ignore_warnings(category=FutureWarning): estimator = clone(estimator_orig)