aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/tools/pip_package/check_load_py_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/tools/pip_package/check_load_py_test.py')
-rw-r--r--tensorflow/tools/pip_package/check_load_py_test.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/tools/pip_package/check_load_py_test.py b/tensorflow/tools/pip_package/check_load_py_test.py
index 79d11b08ce..e2fe1121d7 100644
--- a/tensorflow/tools/pip_package/check_load_py_test.py
+++ b/tensorflow/tools/pip_package/check_load_py_test.py
@@ -22,6 +22,9 @@ import os
import subprocess
+os.chdir(os.path.abspath(os.path.join(os.path.dirname(__file__), '../../..')))
+
+
def check_output_despite_error(args):
"""Get output of args from command line, even if there are errors.