diff options
Diffstat (limited to 'tensorflow/examples/tutorials/estimators/abalone.py')
-rw-r--r-- | tensorflow/examples/tutorials/estimators/abalone.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/examples/tutorials/estimators/abalone.py b/tensorflow/examples/tutorials/estimators/abalone.py index 978af5c9c3..6d8ce2cbc7 100644 --- a/tensorflow/examples/tutorials/estimators/abalone.py +++ b/tensorflow/examples/tutorials/estimators/abalone.py @@ -18,7 +18,7 @@ from __future__ import division from __future__ import print_function import tempfile -import urllib +from six.moves import urllib import numpy as np import tensorflow as tf @@ -51,7 +51,7 @@ def maybe_download(): train_file_name = FLAGS.train_data else: train_file = tempfile.NamedTemporaryFile(delete=False) - urllib.urlretrieve("http://download.tensorflow.org/data/abalone_train.csv", train_file.name) # pylint: disable=line-too-long + urllib.request.urlretrieve("http://download.tensorflow.org/data/abalone_train.csv", train_file.name) # pylint: disable=line-too-long train_file_name = train_file.name train_file.close() print("Training data is downloaded to %s" % train_file_name) @@ -60,7 +60,7 @@ def maybe_download(): test_file_name = FLAGS.test_data else: test_file = tempfile.NamedTemporaryFile(delete=False) - urllib.urlretrieve("http://download.tensorflow.org/data/abalone_test.csv", test_file.name) # pylint: disable=line-too-long + urllib.request.urlretrieve("http://download.tensorflow.org/data/abalone_test.csv", test_file.name) # pylint: disable=line-too-long test_file_name = test_file.name test_file.close() print("Test data is downloaded to %s" % test_file_name) @@ -69,7 +69,7 @@ def maybe_download(): predict_file_name = FLAGS.predict_data else: predict_file = tempfile.NamedTemporaryFile(delete=False) - urllib.urlretrieve("http://download.tensorflow.org/data/abalone_predict.csv", predict_file.name) # pylint: disable=line-too-long + urllib.request.urlretrieve("http://download.tensorflow.org/data/abalone_predict.csv", predict_file.name) # pylint: disable=line-too-long predict_file_name = predict_file.name predict_file.close() print("Prediction data is downloaded to %s" % predict_file_name) |