diff options
author | Michael Case <mikecase@google.com> | 2018-08-09 15:59:47 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-09 16:03:27 -0700 |
commit | 243f6e636c93d63884c574ae9b61d397726189ed (patch) | |
tree | dae4b528389bbf1d99647bdbf69fa340d2ef96c9 /tensorflow/contrib/predictor | |
parent | e1b2dcf381cfea41f1ddd2c674d12f3205391c70 (diff) |
Fix estimator dependencies in contrib/.
PiperOrigin-RevId: 208126204
Diffstat (limited to 'tensorflow/contrib/predictor')
-rw-r--r-- | tensorflow/contrib/predictor/BUILD | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/tensorflow/contrib/predictor/BUILD b/tensorflow/contrib/predictor/BUILD index 36e21af618..72ea777ca7 100644 --- a/tensorflow/contrib/predictor/BUILD +++ b/tensorflow/contrib/predictor/BUILD @@ -60,7 +60,7 @@ py_library( ":base_predictor", "//tensorflow/python:framework_ops", "//tensorflow/python:training", - "//tensorflow/python/estimator:model_fn", + "//tensorflow/python/estimator:estimator_py", "//tensorflow/python/saved_model:signature_constants", ], ) @@ -90,9 +90,7 @@ py_library( "//tensorflow/python:framework_ops", "//tensorflow/python:math_ops", "//tensorflow/python/estimator", - "//tensorflow/python/estimator:export", - "//tensorflow/python/estimator:export_output", - "//tensorflow/python/estimator:model_fn", + "//tensorflow/python/estimator:estimator_py", "//tensorflow/python/saved_model:signature_constants", ], ) |