aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/predictor/BUILD
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/contrib/predictor/BUILD')
-rw-r--r--tensorflow/contrib/predictor/BUILD11
1 files changed, 11 insertions, 0 deletions
diff --git a/tensorflow/contrib/predictor/BUILD b/tensorflow/contrib/predictor/BUILD
index 82cd7b4c8a..d7c3d6c3be 100644
--- a/tensorflow/contrib/predictor/BUILD
+++ b/tensorflow/contrib/predictor/BUILD
@@ -137,6 +137,17 @@ py_test(
)
py_test(
+ name = "predictor_factories_test",
+ srcs = ["predictor_factories_test.py"],
+ data = [":test_export_dir"],
+ srcs_version = "PY2AND3",
+ tags = ["no_pip"],
+ deps = [
+ ":predictor_factories",
+ ],
+)
+
+py_test(
name = "core_estimator_predictor_test",
srcs = ["core_estimator_predictor_test.py"],
srcs_version = "PY2AND3",