aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/tools/pip_package/pip_smoke_test.py
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2017-04-19 10:52:51 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-04-19 12:08:38 -0700
commit721c48c611058aa2d43f7d3b99f5b445741765be (patch)
tree0cd10be679e070c3609f5a3676391317dd17cc56 /tensorflow/tools/pip_package/pip_smoke_test.py
parentc0f6357c4a080edb10dd089151dd523834ea80fc (diff)
Expand pip_smoke test to contrib.
Change: 153614629
Diffstat (limited to 'tensorflow/tools/pip_package/pip_smoke_test.py')
-rw-r--r--tensorflow/tools/pip_package/pip_smoke_test.py14
1 files changed, 12 insertions, 2 deletions
diff --git a/tensorflow/tools/pip_package/pip_smoke_test.py b/tensorflow/tools/pip_package/pip_smoke_test.py
index 10ab871b02..8de639e855 100644
--- a/tensorflow/tools/pip_package/pip_smoke_test.py
+++ b/tensorflow/tools/pip_package/pip_smoke_test.py
@@ -31,7 +31,8 @@ PIP_PACKAGE_QUERY = """bazel query \
PY_TEST_QUERY = """bazel query 'filter("^((?!(benchmark|manual|no_pip)).)*$", \
deps(kind(py_test,\
//tensorflow/python/... + \
- //tensorflow/tensorboard/...), 1))'"""
+ //tensorflow/tensorboard/... + \
+ //tensorflow/contrib/...), 1))'"""
# Hard-coded blacklist of files if not included in pip package
# TODO(amitpatankar): Clean up blacklist.
@@ -43,7 +44,16 @@ BLACKLIST = [
"//tensorflow/python:compare_test_proto_py",
"//tensorflow/core:image_testdata",
"//tensorflow/core/kernels/cloud:bigquery_reader_ops",
- "//tensorflow/python:framework/test_file_system.so"
+ "//tensorflow/python:framework/test_file_system.so",
+ # contrib
+ "//tensorflow/contrib/session_bundle:session_bundle_half_plus_two",
+ "//tensorflow/contrib/keras:testing_utils",
+ "//tensorflow/contrib/ffmpeg:test_data",
+ "//tensorflow/contrib/factorization/examples:mnist",
+ "//tensorflow/contrib/factorization/examples:mnist.py",
+ "//tensorflow/contrib/factorization:factorization_py_CYCLIC_DEPENDENCIES_THAT_NEED_TO_GO", # pylint:disable=line-too-long
+ "//tensorflow/contrib/bayesflow:reinforce_simple_example",
+ "//tensorflow/contrib/bayesflow:examples/reinforce_simple/reinforce_simple_example.py" # pylint:disable=line-too-long
]