Update tests for timm 1.0.16

Signed-off-by: Beat Buesser <[email protected]>

Author: beat-buesser

tests/estimators/certification/test_vision_transformers.py

@@ -549,7 +549,7 @@ def embedder(cls, x, pos_embed, cls_token):
                 x = torch.cat((cls_token.expand(x.shape[0], -1, -1), x), dim=1)
                 return x + pos_embed
 
-        def forward_features(self, x: torch.Tensor) -> torch.Tensor:
+        def forward_features(self, x: torch.Tensor, attn_mask: torch.Tensor | None = None) -> torch.Tensor:
             """
             This is a copy of the function in ArtViT.forward_features
             except we also perform an equivalence assertion compared to the implementation
@@ -558,6 +558,7 @@ def forward_features(self, x: torch.Tensor) -> torch.Tensor:
             The forward pass of the ViT.
 
             :param x: Input data.
+            :param attn_mask: Attention mask.
             :return: The input processed by the ViT backbone
             """
             import copy