Merge pull request #1268 from perib/mate_operator_fix

Mate operator fix
EpistasisLab · Feb 23, 2024 · 7bc629a · 7bc629a
2 parents 3fd4e9b + c6d581f
commit 7bc629a
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),