Skip to content

Commit

Permalink
minor fix on typing
Browse files Browse the repository at this point in the history
  • Loading branch information
maurapintor committed Sep 26, 2024
1 parent 2b877f9 commit df9e4cb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/secmlt/adv/evasion/modular_attack.py
Original file line number Diff line number Diff line change
Expand Up @@ -133,7 +133,7 @@ def _run(
labels: torch.Tensor,
init_deltas: torch.Tensor = None,
**optim_kwargs,
) -> [torch.Tensor, torch.Tensor]:
) -> tuple[torch.Tensor, torch.Tensor]:
multiplier = 1 if self.y_target is not None else -1
target = (
torch.zeros_like(labels) + self.y_target
Expand Down

0 comments on commit df9e4cb

Please sign in to comment.