aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/estimator/BUILD
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/python/estimator/BUILD')
-rw-r--r--tensorflow/python/estimator/BUILD3
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/python/estimator/BUILD b/tensorflow/python/estimator/BUILD
index 82ea1c5b20..8d02ed83af 100644
--- a/tensorflow/python/estimator/BUILD
+++ b/tensorflow/python/estimator/BUILD
@@ -147,6 +147,7 @@ py_test(
":dnn",
":dnn_testing_utils",
":export_export",
+ ":metric_keys",
":numpy_io",
":pandas_io",
":prediction_keys",
@@ -262,6 +263,7 @@ py_test(
"//tensorflow/python:saver_test_utils",
"//tensorflow/python:session",
"//tensorflow/python:state_ops",
+ "//tensorflow/python:summary",
"//tensorflow/python:training",
"//tensorflow/python:util",
"//tensorflow/python:variables",
@@ -492,6 +494,7 @@ py_test(
"//tensorflow/python:platform",
"//tensorflow/python:sparse_tensor",
"//tensorflow/python:state_ops",
+ "//tensorflow/python:summary",
"//tensorflow/python:training",
"//tensorflow/python:variable_scope",
"//tensorflow/python:variables",