bug fix with mate_operator wrapper

EpistasisLab · Aug 25, 2022 · c6d581f · c6d581f
1 parent 9bca3a9
commit c6d581f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tpot/decorators.py b/tpot/decorators.py
@@ -86,6 +86,8 @@ def check_pipeline(self, *args, **kwargs):
                 # returns tuple of (ind1, ind2)
 
                 expr_tuple = expr if isinstance(expr, tuple) else (expr,)
+                if func.__name__ == "_mate_operator":
+                    expr_tuple = expr_tuple[0:2]
                 for expr_test in expr_tuple:
                     pipeline_code = generate_pipeline_code(
                         expr_to_tree(expr_test, self._pset),