Add dependency for L0_python_client_unit_tests (triton-inference-serv…

…er#6010)
dsgibbons · Jul 5, 2023 · 15dbea1 · 15dbea1
1 parent b95366e
commit 15dbea1
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/qa/L0_python_client_unit_tests/test.sh b/qa/L0_python_client_unit_tests/test.sh
@@ -29,6 +29,9 @@ TEST_LOG="./python_client_unit_tests.log"
 PYTHON_CLIENT_UNIT_TESTS_DIR=/opt/tritonserver/qa/python_client_unit_tests/
 PYTHON_CLIENT_UNIT_TESTS_CMD="python3 -m unittest discover -v -s $PYTHON_CLIENT_UNIT_TESTS_DIR -t $PYTHON_CLIENT_UNIT_TESTS_DIR"
 
+# DLPack test requires Torch to validate GPU tensor
+pip3 install torch
+
 RET=0
 
 rm -f $TEST_LOG