diff options
Diffstat (limited to 'tensorflow/tools/pip_package/pip_smoke_test.py')
-rw-r--r-- | tensorflow/tools/pip_package/pip_smoke_test.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/tensorflow/tools/pip_package/pip_smoke_test.py b/tensorflow/tools/pip_package/pip_smoke_test.py index cddf9c8f44..8eee489e2d 100644 --- a/tensorflow/tools/pip_package/pip_smoke_test.py +++ b/tensorflow/tools/pip_package/pip_smoke_test.py @@ -23,8 +23,13 @@ from __future__ import absolute_import from __future__ import division from __future__ import print_function +import os import subprocess + +os.chdir(os.path.abspath(os.path.join(os.path.dirname(__file__), '../../..'))) + + PIP_PACKAGE_QUERY_EXPRESSION = \ 'deps(//tensorflow/tools/pip_package:build_pip_package)' @@ -134,8 +139,8 @@ def main(): raise RuntimeError("""One or more dependencies are not in the pip package. Please either blacklist the dependencies in -tensorflow/tensorflow/tensorflow/tools/pip_package/pip_smoke_test.py -or add them to tensorflow/tensorflow/tensorflow/tools/pip_package/BUILD.""") +//tensorflow/tools/pip_package/pip_smoke_test.py +or add them to //tensorflow/tools/pip_package/BUILD.""") else: print("TEST PASSED") |