chore: fix pipeline test

zama-ai · Sep 14, 2023 · c141ac4 · c141ac4
1 parent 74e5e24
commit c141ac4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -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(