From 74b83986682bb1515abf40bba158bef853ff6c7e Mon Sep 17 00:00:00 2001 From: Eli Bixby Date: Tue, 21 Mar 2017 11:06:48 -0700 Subject: [PATCH] Coerce string to bytes in py2-3 compat way --- ml_engine/online_prediction/predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ml_engine/online_prediction/predict.py b/ml_engine/online_prediction/predict.py index 8d6bb8fa181..37d18a89b76 100644 --- a/ml_engine/online_prediction/predict.py +++ b/ml_engine/online_prediction/predict.py @@ -122,7 +122,7 @@ def census_to_example_bytes(json_instance): for key, data in six.iteritems(json_instance): if isinstance(data, six.string_types): feature_dict[key] = tf.train.Feature( - bytes_list=tf.train.BytesList(value=[str(data)])) + bytes_list=tf.train.BytesList(value=[data.encode('utf-8')])) elif isinstance(data, float): feature_dict[key] = tf.train.Feature( float_list=tf.train.FloatList(value=[data]))