aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/examples/get_started/regression/test.py
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/examples/get_started/regression/test.py')
-rw-r--r--tensorflow/examples/get_started/regression/test.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/examples/get_started/regression/test.py b/tensorflow/examples/get_started/regression/test.py
index 0b1477ad96..bb4db6700b 100644
--- a/tensorflow/examples/get_started/regression/test.py
+++ b/tensorflow/examples/get_started/regression/test.py
@@ -29,7 +29,7 @@ import tensorflow.examples.get_started.regression.imports85 as imports85
sys.modules["imports85"] = imports85
# pylint: disable=g-bad-import-order,g-import-not-at-top
-import tensorflow.contrib.data as data
+import tensorflow.data as data
import tensorflow.examples.get_started.regression.dnn_regression as dnn_regression
import tensorflow.examples.get_started.regression.linear_regression as linear_regression