diff --git a/qa/TL2_FW_iterators_perf/test_paddle.sh b/qa/TL2_FW_iterators_perf/test_paddle.sh index a6be02de5c..75656a7c0a 100755 --- a/qa/TL2_FW_iterators_perf/test_paddle.sh +++ b/qa/TL2_FW_iterators_perf/test_paddle.sh @@ -1,6 +1,6 @@ #!/bin/bash -e # used pip packages -pip_packages='paddlepaddle-gpu' +pip_packages='paddlepaddle-gpu nose' target_dir=./dali/test/python one_config_only=true diff --git a/qa/TL2_FW_iterators_perf/test_pytorch.sh b/qa/TL2_FW_iterators_perf/test_pytorch.sh index ce5536db13..5df1a4a74e 100755 --- a/qa/TL2_FW_iterators_perf/test_pytorch.sh +++ b/qa/TL2_FW_iterators_perf/test_pytorch.sh @@ -1,6 +1,6 @@ #!/bin/bash -e # used pip packages -pip_packages='pillow torchvision torch opencv-python' +pip_packages='pillow torchvision torch opencv-python nose' target_dir=./dali/test/python one_config_only=true diff --git a/qa/TL2_FW_iterators_perf/test_tf.sh b/qa/TL2_FW_iterators_perf/test_tf.sh index 81f0948cf3..5a958ae337 100755 --- a/qa/TL2_FW_iterators_perf/test_tf.sh +++ b/qa/TL2_FW_iterators_perf/test_tf.sh @@ -1,6 +1,6 @@ #!/bin/bash -e # used pip packages -pip_packages='tensorflow-gpu' +pip_packages='tensorflow-gpu nose' target_dir=./dali/test/python one_config_only=true diff --git a/qa/TL2_RN50_data_perf/test.sh b/qa/TL2_RN50_data_perf/test.sh index 5ab3184330..8fa95864f1 100755 --- a/qa/TL2_RN50_data_perf/test.sh +++ b/qa/TL2_RN50_data_perf/test.sh @@ -1,6 +1,6 @@ #!/bin/bash -e # used pip packages -pip_packages='opencv-python' +pip_packages='opencv-python nose' target_dir=./dali/test/python do_once() {