From 5cd1cc6e5cb31733f45f98590754cd738232fa43 Mon Sep 17 00:00:00 2001 From: kcelia Date: Thu, 14 Sep 2023 14:40:09 +0200 Subject: [PATCH] chore: fix pipeline test --- tests/sklearn/test_sklearn_models.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/sklearn/test_sklearn_models.py b/tests/sklearn/test_sklearn_models.py index d56ceeac9..a59f95a55 100644 --- a/tests/sklearn/test_sklearn_models.py +++ b/tests/sklearn/test_sklearn_models.py @@ -807,7 +807,8 @@ def get_hyper_param_combinations(model_class): "base_score": [0.5, None], } elif model_class in get_sklearn_neighbors_models(): - hyper_param_combinations = {"n_neighbors": [2, 4]} + # Use small `n_neighbors` values ​​for KNN, because the data-set is too small for now + hyper_param_combinations = {"n_neighbors": [2]} else: assert is_model_class_in_a_list(