Skip to content

Commit 5f3ccef

Browse files
authored
Merge pull request #1135 from weixuanfu/master
Fix the logic of type checking in operator_utils
2 parents 08e9db8 + 7f6a23e commit 5f3ccef

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

tpot/_version.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,4 +23,4 @@
2323
2424
"""
2525

26-
__version__ = '0.11.6'
26+
__version__ = '0.11.6.post1'

tpot/operator_utils.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -194,10 +194,11 @@ def TPOTOperatorClassFactory(opsourse, opdict, BaseClass=Operator, ArgBaseClass=
194194
elif is_regressor(op_obj):
195195
class_profile['root'] = True
196196
optype = "Regressor"
197-
if _is_transformer(op_obj):
198-
optype = "Transformer"
199-
if _is_selector(op_obj):
197+
elif _is_selector(op_obj):
200198
optype = "Selector"
199+
elif _is_transformer(op_obj):
200+
optype = "Transformer"
201+
201202

202203
@classmethod
203204
def op_type(cls):

0 commit comments

Comments
 (0)