Skip to content

Commit e61c304

Browse files
JanuszLklecki
authored andcommitted
Fix problem with TensorFlow and cupy tests (#1568)
- when tensorflow-gpu v1.15.0 is installed in the conda environment some strange dependency leads to `ModuleNotFoundError: No module named 'prompt_toolkit.formatted_text'` error - according to jupyter/notebook#4050 installing `ipykernel<5.0.0 ipython<7.0.0` helps - fixes cupy version to 6.6.0 as this the last one supporting python 2.7 Signed-off-by: Janusz Lisiecki <[email protected]>
1 parent ff59a82 commit e61c304

File tree

3 files changed

+3
-5
lines changed

3 files changed

+3
-5
lines changed

qa/TL0_python_self_test_frameworks/test_cupy.sh

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,7 @@ pip_packages="nose numpy cupy"
44
target_dir=./dali/test/python
55

66
test_body() {
7-
if [[ ${PYTHON_VERSION} != 2.7 ]]
8-
then
9-
nosetests --verbose -m '(?:^|[\b_\./-])[Tt]est.*cupy' test_dltensor_operator.py
10-
fi
7+
nosetests --verbose -m '(?:^|[\b_\./-])[Tt]est.*cupy' test_dltensor_operator.py
118
}
129

1310
pushd ../..

qa/TL1_tensorflow_dataset/test.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
#!/bin/bash -e
22
# used pip packages
3-
pip_packages="nose jupyter tensorflow-gpu"
3+
pip_packages="nose jupyter tensorflow-gpu ipykernel<5.0.0 ipython<7.0.0"
44
target_dir=./dali/test/python
55

66
# populate epilog and prolog with variants to enable/disable conda and virtual env

qa/setup_packages.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222

2323
packages = {
2424
"opencv-python" : ["4.1.0.25"],
25+
"cupy" : ["6.6.0"],
2526
"mxnet-cu{cuda_v}" : {
2627
"90" : ["1.5.0"],
2728
"100" : ["1.5.0"]},

0 commit comments

Comments
 (0)