File tree Expand file tree Collapse file tree 2 files changed +5
-2
lines changed Expand file tree Collapse file tree 2 files changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -118,6 +118,8 @@ def serialize_possible_ml_model(possible_ml_model):
118
118
:type possible_ml_model: obj
119
119
:return: tuple(bool, whether itself or dict)
120
120
"""
121
+ if isinstance (possible_ml_model , str ):
122
+ return False , possible_ml_model
121
123
ml_category , transporter = get_transporter (possible_ml_model )
122
124
if transporter is not None :
123
125
return True , {
Original file line number Diff line number Diff line change @@ -29,8 +29,9 @@ def get_concrete_transporter(model):
29
29
:return: tuple(ML_MODEL_CATEGORY, transporter function)
30
30
"""
31
31
if isinstance (model , str ):
32
- if model .upper () in MODEL_TYPE_TRANSPORTER .keys ():
33
- return model , MODEL_TYPE_TRANSPORTER [model ]
32
+ upper_model = model .upper ()
33
+ if upper_model in MODEL_TYPE_TRANSPORTER .keys ():
34
+ return upper_model , MODEL_TYPE_TRANSPORTER [upper_model ]
34
35
35
36
if is_linear_model (model ):
36
37
return "LINEAR_MODEL" , transport_linear_model
You can’t perform that action at this time.
0 commit comments