diff --git a/qa/L0_backend_python/python_unittest.py b/qa/L0_backend_python/python_unittest.py index 6702d06b3a..9ff1b30e02 100644 --- a/qa/L0_backend_python/python_unittest.py +++ b/qa/L0_backend_python/python_unittest.py @@ -26,7 +26,7 @@ import sys -sys.path.append("/opt/tritonserver/qa/common") +sys.path.append("../../common") import test_util as tu import shm_util diff --git a/qa/L0_warmup/test.sh b/qa/L0_warmup/test.sh index ecc659dbcf..193f4b130d 100644 --- a/qa/L0_warmup/test.sh +++ b/qa/L0_warmup/test.sh @@ -42,7 +42,7 @@ export CUDA_VISIBLE_DEVICES=0 CLIENT=../clients/image_client CLIENT_LOG="./client.log" -CLIENT_PY=/opt/tritonserver/qa/L0_backend_python/python_unittest.py +CLIENT_PY=./python_unittest.py EXPECTED_NUM_TESTS="1" TEST_RESULT_FILE='test_results.txt' @@ -449,6 +449,9 @@ mkdir -p models/bls_onnx_warmup/1/ cp ../python_models/bls_onnx_warmup/model.py models/bls_onnx_warmup/1/ cp ../python_models/bls_onnx_warmup/config.pbtxt models/bls_onnx_warmup/. +cp ../L0_backend_python/python_unittest.py . +sed -i 's#sys.path.append("../../common")#sys.path.append("../common")#g' python_unittest.py + run_server if [ "$SERVER_PID" == "0" ]; then echo -e "\n***\n*** Failed to start $SERVER\n***"