From d11ad089c24f656f3053161b33c1eeb06ec8799d Mon Sep 17 00:00:00 2001 From: Janis Klaise Date: Mon, 11 Feb 2019 16:15:20 +0000 Subject: [PATCH] Fix bug in parsing boolean params in Python wrapper --- python/seldon_core/microservice.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/python/seldon_core/microservice.py b/python/seldon_core/microservice.py index 698cbdfbb0..5f7c021a3e 100644 --- a/python/seldon_core/microservice.py +++ b/python/seldon_core/microservice.py @@ -12,6 +12,7 @@ from google.protobuf import json_format from google.protobuf.struct_pb2 import ListValue import sys +from distutils.util import strtobool from seldon_core.proto import prediction_pb2 import seldon_core.persistence as persistence @@ -243,7 +244,10 @@ def parse_parameters(parameters): name = param.get("name") value = param.get("value") type_ = param.get("type") - parsed_parameters[name] = type_dict[type_](value) + if type_ == "BOOL": + parsed_parameters[name] = bool(strtobool(value)) + else: + parsed_parameters[name] = type_dict[type_](value) return parsed_parameters