Add nose as a dependency to iterators tests (#5433)

Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>
NVIDIA · Apr 18, 2024 · 1bc7fc2 · 1bc7fc2
1 parent 4260a11
commit 1bc7fc2
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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() {