Skip to content

Commit 85a0896

Browse files
committed
Update test_LONG_search_images_keras.py
1 parent 1f6aa68 commit 85a0896

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

_unittests/ut_search_rank/test_LONG_search_images_keras.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,9 +28,9 @@ def test_search_predictions_keras(self):
2828
with redirect_stderr(StringIO()):
2929
try:
3030
from keras.applications.mobilenet import MobileNet # pylint: disable=E0401,E0611
31-
except (SyntaxError, ModuleNotFoundError) as e:
31+
except (SyntaxError, ModuleNotFoundError, AttributeError) as e:
3232
warnings.warn(
33-
"tensorflow is probably not available yet on python 3.7: {0}".format(e))
33+
"Issue with tensorflow or keras: {0}".format(e))
3434
return
3535
from keras.preprocessing.image import ImageDataGenerator # pylint: disable=E0401,E0611
3636
from keras.preprocessing.image import img_to_array, load_img # pylint: disable=E0401,E0611

0 commit comments

Comments
 (0)