Skip to content

Commit d17a063

Browse files
author
Beat Buesser
committed
Update type of delta
Signed-off-by: Beat Buesser <[email protected]>
1 parent b8e539b commit d17a063

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

art/attacks/evasion/feature_adversaries/feature_adversaries_pytorch.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
| Paper link: https://arxiv.org/abs/1511.05122
2222
"""
2323
import logging
24-
from typing import TYPE_CHECKING, Optional, Union
24+
from typing import TYPE_CHECKING, Optional, Tuple, Union
2525

2626
import numpy as np
2727
from tqdm.auto import trange
@@ -68,7 +68,7 @@ def __init__(
6868
optimizer: Optional["Optimizer"] = None,
6969
optimizer_kwargs: Optional[dict] = None,
7070
lambda_: float = 0.0,
71-
layer: Union[int, str] = -1,
71+
layer: Union[int, str, Tuple[int, ...], Tuple[str, ...]] = -1,
7272
max_iter: int = 100,
7373
batch_size: int = 32,
7474
step_size: Optional[Union[int, float]] = None,
@@ -93,9 +93,9 @@ def __init__(
9393
"""
9494
super().__init__(estimator=estimator)
9595

96+
self.delta = delta
9697
self.optimizer = optimizer
9798
self._optimizer_kwargs = {} if optimizer_kwargs is None else optimizer_kwargs
98-
self.delta = delta
9999
self.lambda_ = lambda_
100100
self.layer = layer if isinstance(layer, tuple) else (layer,)
101101
self.batch_size = batch_size

0 commit comments

Comments
 (0)