From a286b7db8ee5e9f496099b9d9ea27dcb5f9efe9e Mon Sep 17 00:00:00 2001 From: Yifei Feng Date: Wed, 21 Jun 2017 11:19:51 -0700 Subject: [PATCH] Make debug_test slice integer. --- .../contrib/learn/python/learn/estimators/debug_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/contrib/learn/python/learn/estimators/debug_test.py b/tensorflow/contrib/learn/python/learn/estimators/debug_test.py index 935e66ee8ca..ab5db3a151a 100644 --- a/tensorflow/contrib/learn/python/learn/estimators/debug_test.py +++ b/tensorflow/contrib/learn/python/learn/estimators/debug_test.py @@ -52,8 +52,8 @@ LABEL_DIMENSION = 3 # Dimensionality of regression labels. def _train_test_split(features_and_labels): features, labels = features_and_labels - train_set = (features[:len(features) / 2], labels[:len(features) / 2]) - test_set = (features[len(features) / 2:], labels[len(features) / 2:]) + train_set = (features[:int(len(features) / 2)], labels[:int(len(features) / 2)]) + test_set = (features[int(len(features) / 2):], labels[int(len(features) / 2):]) return train_set, test_set