Merge pull request #1135 from weixuanfu/master

Fix the logic of type checking in operator_utils
EpistasisLab · Nov 5, 2020 · 5f3ccef · 5f3ccef
2 parents 08e9db8 + 7f6a23e
commit 5f3ccef
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/tpot/_version.py b/tpot/_version.py
@@ -23,4 +23,4 @@
 
 """
 
-__version__ = '0.11.6'
+__version__ = '0.11.6.post1'
diff --git a/tpot/operator_utils.py b/tpot/operator_utils.py
@@ -194,10 +194,11 @@ def TPOTOperatorClassFactory(opsourse, opdict, BaseClass=Operator, ArgBaseClass=
         elif is_regressor(op_obj):
             class_profile['root'] = True
             optype = "Regressor"
-        if _is_transformer(op_obj):
-            optype = "Transformer"
-        if _is_selector(op_obj):
+        elif _is_selector(op_obj):
             optype = "Selector"
+        elif _is_transformer(op_obj):
+            optype = "Transformer"
+
 
         @classmethod
         def op_type(cls):