test warm_start_configs

rodrigo-arenas · Sep 12, 2024 · 265cda8 · 265cda8
1 parent a84e7ba
commit 265cda8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/sklearn_genetic/space/space.py b/sklearn_genetic/space/space.py
@@ -240,7 +240,7 @@ def sample_warm_start(self, warm_start_values: dict):
             if param in warm_start_values:
                 sampled_params[param] = warm_start_values[param]
             else:
-                sampled_params[param] = dimension.sample()  # Random sample if no warm-start value
+                sampled_params[param] = dimension.sample()
         return sampled_params
 
     @property

diff --git a/sklearn_genetic/tests/test_genetic_search.py b/sklearn_genetic/tests/test_genetic_search.py
@@ -605,6 +605,8 @@ def test_expected_ga_schedulers():
             "average": Categorical([True, False]),
             "max_iter": Integer(700, 1000),
         },
+        warm_start_configs=[{"l1_ratio": 0.5, "alpha": 0.5, "average": False, "max_iter": 400},
+                            {"l1_ratio": 0.2, "alpha": 0.8, "average": True, "max_iter": 400}],
         verbose=False,
         algorithm="eaSimple",
         n_jobs=-1,