From 75bb036d8d47086731c36d862d126335c69b34dc Mon Sep 17 00:00:00 2001 From: jeremie du boisberranger Date: Fri, 22 Apr 2022 14:16:38 +0200 Subject: [PATCH] speed-up test --- sklearn/decomposition/tests/test_dict_learning.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/sklearn/decomposition/tests/test_dict_learning.py b/sklearn/decomposition/tests/test_dict_learning.py index 7eef4f480d7de..f064094c78f95 100644 --- a/sklearn/decomposition/tests/test_dict_learning.py +++ b/sklearn/decomposition/tests/test_dict_learning.py @@ -911,9 +911,12 @@ def test_minibatch_dictionary_learning_dtype_match( batch_size=10, fit_algorithm=fit_algorithm, transform_algorithm=transform_algorithm, + max_iter=100, + tol=1e-1, random_state=0, ) dict_learner.fit(X.astype(data_type)) + assert dict_learner.components_.dtype == expected_type assert dict_learner.transform(X.astype(data_type)).dtype == expected_type assert dict_learner._inner_stats[0].dtype == expected_type