minor fixes

aimclub · Aug 10, 2023 · 5341a33 · 5341a33
1 parent 2b6cae9
commit 5341a33
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 16 deletions.
diff --git a/bamt/builders/composite_builder.py b/bamt/builders/composite_builder.py
@@ -187,21 +187,24 @@ def search(self, data: DataFrame, **kwargs) -> [List[Tuple[str, str]], Dict]:
         preprocessed_data, _ = p.apply(data)
 
         # Create the initial population
-        initial = [
-            CompositeModel(
-                nodes=[
-                    CompositeNode(
-                        nodes_from=None,
-                        content={
-                            "name": vertex,
-                            "type": p.nodes_types[vertex],
-                            "parent_model": None,
-                        },
-                    )
-                    for vertex in vertices
-                ]
-            )
-        ]
+        initial = kwargs.get(
+            "custom_initial_structure",
+            [
+                CompositeModel(
+                    nodes=[
+                        CompositeNode(
+                            nodes_from=None,
+                            content={
+                                "name": vertex,
+                                "type": p.nodes_types[vertex],
+                                "parent_model": None,
+                            },
+                        )
+                        for vertex in vertices
+                    ]
+                )
+            ],
+        )
 
         # Define the requirements for the evolutionary algorithm
         requirements = GraphRequirements(

diff --git a/bamt/builders/evo_builder.py b/bamt/builders/evo_builder.py
@@ -79,7 +79,6 @@ def __init__(
         self.default_timeout = 180
         self.default_num_of_generations = 50
         self.default_early_stopping_iterations = 50
-        self.verbose = (True,)
         self.logging_level = 50
         self.objective_metric = evo.K2_metric
         self.default_crossovers = [
@@ -98,6 +97,7 @@ def __init__(
             has_no_cycle,
             evo.has_no_duplicates,
         ]
+        self.verbose = True
 
     def build(
         self,