add creator

rodrigo-arenas · Sep 17, 2024 · 11c6cce · 11c6cce
1 parent b22e550
commit 11c6cce
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sklearn_genetic/genetic_search.py b/sklearn_genetic/genetic_search.py
@@ -263,7 +263,7 @@ def __init__(
         self.pre_dispatch = pre_dispatch
         self.error_score = error_score
         self.return_train_score = return_train_score
-        # self.creator = creator
+        self.creator = creator
         self.log_config = log_config
         self.use_cache = use_cache
         self.fitness_cache = {}
@@ -939,7 +939,7 @@ def __init__(
         self.pre_dispatch = pre_dispatch
         self.error_score = error_score
         self.return_train_score = return_train_score
-        # self.creator = creator
+        self.creator = creator
         self.log_config = log_config
         self.use_cache = use_cache
         self.fitness_cache = {}