Skip to content

Commit

Permalink
Unpin librosa as it currently causes an issue about: cannot allocate …
Browse files Browse the repository at this point in the history
…memory in static TLS block

Signed-off-by: Joaquin Anton <janton@nvidia.com>
  • Loading branch information
jantonguirao committed Jun 7, 2024
1 parent ebfb155 commit 6f890c4
Show file tree
Hide file tree
Showing 13 changed files with 13 additions and 13 deletions.
2 changes: 1 addition & 1 deletion qa/TL0_cpu_only/test_nofw.sh
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
#!/bin/bash -e
# used pip packages
pip_packages='${python_test_runner_package} numpy numba scipy librosa==0.8.1'
pip_packages='${python_test_runner_package} numpy numba scipy librosa'

target_dir=./dali/test/python

Expand Down
2 changes: 1 addition & 1 deletion qa/TL0_cpu_only/test_pytorch.sh
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
#!/bin/bash -e
# used pip packages
pip_packages='${python_test_runner_package} numpy pillow torch numba scipy librosa==0.8.1'
pip_packages='${python_test_runner_package} numpy pillow torch numba scipy librosa'

target_dir=./dali/test/python

Expand Down
2 changes: 1 addition & 1 deletion qa/TL0_jupyter/test_nofw.sh
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
#!/bin/bash -e
# used pip packages
pip_packages='jupyter numpy matplotlib<3.5.3 pillow opencv-python librosa==0.8.1 simpleaudio'
pip_packages='jupyter numpy matplotlib<3.5.3 pillow opencv-python librosa simpleaudio'
target_dir=./docs/examples

do_once() {
Expand Down
2 changes: 1 addition & 1 deletion qa/TL0_multigpu/test_nofw.sh
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
#!/bin/bash -e
# used pip packages
pip_packages='${python_test_runner_package} numpy librosa==0.8.1 scipy nvidia-ml-py==11.450.51 psutil dill cloudpickle opencv-python pillow'
pip_packages='${python_test_runner_package} numpy librosa scipy nvidia-ml-py==11.450.51 psutil dill cloudpickle opencv-python pillow'
target_dir=./dali/test/python

# test_body definition is in separate file so it can be used without setup
Expand Down
2 changes: 1 addition & 1 deletion qa/TL0_python-self-test-core/test_nofw.sh
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
#!/bin/bash -e
# used pip packages
pip_packages='${python_test_runner_package} numpy opencv-python pillow nvidia-ml-py==11.450.51 numba lz4 scipy mypy librosa==0.8.1'
pip_packages='${python_test_runner_package} numpy opencv-python pillow nvidia-ml-py==11.450.51 numba lz4 scipy mypy librosa'

target_dir=./dali/test/python

Expand Down
2 changes: 1 addition & 1 deletion qa/TL0_python-self-test-operators_1/test_nofw.sh
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
#!/bin/bash -e
# used pip packages
pip_packages='${python_test_runner_package} dataclasses numpy opencv-python pillow librosa==0.8.1 scipy nvidia-ml-py==11.450.51 numba lz4'
pip_packages='${python_test_runner_package} dataclasses numpy opencv-python pillow librosa scipy nvidia-ml-py==11.450.51 numba lz4'

target_dir=./dali/test/python

Expand Down
2 changes: 1 addition & 1 deletion qa/TL0_python-self-test-operators_2/test_nofw.sh
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
#!/bin/bash -e
# used pip packages
pip_packages='${python_test_runner_package} dataclasses numpy opencv-python pillow librosa==0.8.1 scipy nvidia-ml-py==11.450.51 numba lz4'
pip_packages='${python_test_runner_package} dataclasses numpy opencv-python pillow librosa scipy nvidia-ml-py==11.450.51 numba lz4'

target_dir=./dali/test/python

Expand Down
2 changes: 1 addition & 1 deletion qa/TL0_python-self-test-readers-decoders/test_nofw.sh
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
#!/bin/bash -e
# used pip packages
pip_packages='${python_test_runner_package} numpy librosa==0.8.1 scipy nvidia-ml-py==11.450.51 psutil dill cloudpickle pillow opencv-python astropy'
pip_packages='${python_test_runner_package} numpy librosa scipy nvidia-ml-py==11.450.51 psutil dill cloudpickle pillow opencv-python astropy'

target_dir=./dali/test/python

Expand Down
2 changes: 1 addition & 1 deletion qa/TL0_python_self_test_frameworks/test_pytorch.sh
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
#!/bin/bash -e
# used pip packages
pip_packages='${python_test_runner_package} numpy librosa==0.8.1 torch psutil'
pip_packages='${python_test_runner_package} numpy librosa torch psutil'
target_dir=./dali/test/python

test_body() {
Expand Down
2 changes: 1 addition & 1 deletion qa/TL0_python_self_test_frameworks/test_pytorch_cupy.sh
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
#!/bin/bash -e
# used pip packages
pip_packages='${python_test_runner_package} numpy librosa==0.8.1 torch psutil cupy'
pip_packages='${python_test_runner_package} numpy librosa torch psutil cupy'
target_dir=./dali/test/python

test_body() {
Expand Down
2 changes: 1 addition & 1 deletion qa/TL1_jupyter_conda/test_nofw.sh
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
#!/bin/bash -e
# used pip packages
pip_packages='jupyter numpy matplotlib<3.5.3 pillow opencv-python librosa==0.8.1 simpleaudio'
pip_packages='jupyter numpy matplotlib<3.5.3 pillow opencv-python librosa simpleaudio'
target_dir=./docs/examples

# populate epilog and prolog with variants to enable/disable conda
Expand Down
2 changes: 1 addition & 1 deletion qa/TL1_python-self-test-slow/test.sh
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
#!/bin/bash -e
# used pip packages
pip_packages='${python_test_runner_package} dataclasses numpy opencv-python pillow librosa==0.8.1 scipy nvidia-ml-py==11.450.51 numba lz4'
pip_packages='${python_test_runner_package} dataclasses numpy opencv-python pillow librosa scipy nvidia-ml-py==11.450.51 numba lz4'
target_dir=./dali/test/python

test_body() {
Expand Down
2 changes: 1 addition & 1 deletion qa/TL1_python-self-test_conda/test_nofw.sh
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
#!/bin/bash -e
# used pip packages
pip_packages='${python_test_runner_package} dataclasses numpy opencv-python pillow librosa==0.8.1 scipy nvidia-ml-py==11.450.51 numba lz4 psutil dill cloudpickle astropy'
pip_packages='${python_test_runner_package} dataclasses numpy opencv-python pillow librosa scipy nvidia-ml-py==11.450.51 numba lz4 psutil dill cloudpickle astropy'
target_dir=./dali/test/python

# test_body definition is in separate file so it can be used without setup
Expand Down

0 comments on commit 6f890c4

Please sign in to comment.