Skip to content

Commit 00a8e5e

Browse files
author
Beat Buesser
committed
Check scipy version for PixelThreshold attack
Signed-off-by: Beat Buesser <[email protected]>
1 parent bdab55a commit 00a8e5e

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

art/attacks/evasion/pixel_threshold.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,8 +39,14 @@
3939
# Otherwise may use Tensorflow's implementation of DE.
4040

4141
from six import string_types
42+
import scipy
4243
from scipy._lib._util import check_random_state
43-
from scipy.optimize.optimize import _status_message
44+
45+
scipy_version = list(map(int, scipy.__version__.lower().split(".")))
46+
if scipy_version[1] >= 8:
47+
from scipy.optimize._optimize import _status_message
48+
else:
49+
from scipy.optimize.optimize import _status_message
4450
from scipy.optimize import OptimizeResult, minimize
4551
from tqdm.auto import tqdm
4652

0 commit comments

Comments
 (0)