aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/learn
diff options
context:
space:
mode:
authorGravatar Michael Case <mikecase@google.com>2018-08-09 15:59:47 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-08-09 16:03:27 -0700
commit243f6e636c93d63884c574ae9b61d397726189ed (patch)
treedae4b528389bbf1d99647bdbf69fa340d2ef96c9 /tensorflow/contrib/learn
parente1b2dcf381cfea41f1ddd2c674d12f3205391c70 (diff)
Fix estimator dependencies in contrib/.
PiperOrigin-RevId: 208126204
Diffstat (limited to 'tensorflow/contrib/learn')
-rw-r--r--tensorflow/contrib/learn/BUILD19
1 files changed, 5 insertions, 14 deletions
diff --git a/tensorflow/contrib/learn/BUILD b/tensorflow/contrib/learn/BUILD
index b56a88659b..d3aa3fa92c 100644
--- a/tensorflow/contrib/learn/BUILD
+++ b/tensorflow/contrib/learn/BUILD
@@ -79,16 +79,7 @@ py_library(
"//tensorflow/python:variable_scope",
"//tensorflow/python:variables",
"//tensorflow/python:weights_broadcast_ops",
- "//tensorflow/python/estimator",
"//tensorflow/python/estimator:estimator_py",
- "//tensorflow/python/estimator:export_export",
- "//tensorflow/python/estimator:export_output",
- "//tensorflow/python/estimator:inputs",
- "//tensorflow/python/estimator:inputs_queues",
- "//tensorflow/python/estimator:model_fn",
- "//tensorflow/python/estimator:numpy_io",
- "//tensorflow/python/estimator:pandas_io",
- "//tensorflow/python/estimator:run_config",
"//tensorflow/python/feature_column",
"//tensorflow/python/feature_column:feature_column_py",
"//tensorflow/python/ops/losses",
@@ -171,7 +162,7 @@ tf_py_test(
"//tensorflow/python:training",
"//tensorflow/python:util",
"//tensorflow/python:variables",
- "//tensorflow/python/estimator",
+ "//tensorflow/python/estimator:estimator_py",
],
tags = ["no_windows"], # TODO: needs investigation on Windows
)
@@ -220,7 +211,7 @@ py_test(
"//tensorflow/contrib/training:training_py",
"//tensorflow/python:client_testlib",
"//tensorflow/python:platform",
- "//tensorflow/python/estimator:run_config",
+ "//tensorflow/python/estimator:estimator_py",
],
)
@@ -245,7 +236,7 @@ py_test(
"//tensorflow/python:summary",
"//tensorflow/python:training",
"//tensorflow/python:variables",
- "//tensorflow/python/estimator",
+ "//tensorflow/python/estimator:estimator_py",
],
)
@@ -259,7 +250,7 @@ py_test(
"//tensorflow/core:protos_all_py",
"//tensorflow/python:client_testlib",
"//tensorflow/python:training",
- "//tensorflow/python/estimator:run_config",
+ "//tensorflow/python/estimator:estimator_py",
],
)
@@ -621,7 +612,7 @@ py_test(
"//tensorflow/python:control_flow_ops",
"//tensorflow/python:session",
"//tensorflow/python:training",
- "//tensorflow/python/estimator:export_output",
+ "//tensorflow/python/estimator:estimator_py",
"//tensorflow/python/saved_model:signature_constants",
"@six_archive//:six",
],