From c6d581fb7f67b13330b363e769603570a9560483 Mon Sep 17 00:00:00 2001 From: perib Date: Thu, 25 Aug 2022 14:27:59 -0700 Subject: [PATCH] bug fix with mate_operator wrapper --- tpot/decorators.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tpot/decorators.py b/tpot/decorators.py index 3dac4b3c..1e037cf0 100644 --- 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),