Skip to content

Commit

Permalink
fix enseble data typing due to pytorch update
Browse files Browse the repository at this point in the history
  • Loading branch information
maurapintor committed Nov 2, 2024
1 parent 753b1d6 commit 1e7f856
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/secmlt/adv/evasion/aggregators/ensemble.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ def __call__(
model: BaseModel,
data_loader: DataLoader,
adv_loaders: list[DataLoader],
) -> DataLoader[torch.Tuple[torch.Tensor]]:
) -> DataLoader[tuple[torch.Tensor]]:
"""
Get the worst-case of the metric with the given implemented criterion.
Expand Down

0 comments on commit 1e7f856

Please sign in to comment.