From c141ac4d463cd02f78b9d5cf7065a0299124fd73 Mon Sep 17 00:00:00 2001 From: kcelia Date: Thu, 14 Sep 2023 14:40:09 +0200 Subject: [PATCH] chore: fix pipeline test --- conftest.py | 2 +- tests/sklearn/test_sklearn_models.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/conftest.py b/conftest.py index 507c7742d..2183d9d9f 100644 --- a/conftest.py +++ b/conftest.py @@ -151,7 +151,7 @@ def default_configuration_impl(mode: Optional[str] = "MULTI"): """Return the default test compilation configuration for a given strategy. Args: - mode (Optional[str]): Wether we execute in a MONO or MULTI precision settings. + mode (Optional[str]): Whether we execute in a MONO or MULTI precision settings. Returns: The configuration. diff --git a/tests/sklearn/test_sklearn_models.py b/tests/sklearn/test_sklearn_models.py index d56ceeac9..caad6fdf7 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(