Skip to content

Commit 7f6a23e

Browse files
weixuanf@upenn.eduweixuanf@upenn.edu
authored andcommitted
fix a logic type checking in operator_utils
1 parent 08e9db8 commit 7f6a23e

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)