Mypy updates for torch 2.8.0

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

Author: beat-buesser

art/estimators/certification/deep_z/pytorch.py

@@ -169,7 +169,7 @@ def concrete_forward(self, in_x: np.ndarray | "torch.Tensor") -> "torch.Tensor":
             # as reshapes are not modules we infer when the reshape from convolutional to dense occurs
             if self.reshape_op_num == op_num:
                 x = x.reshape((x.shape[0], -1))
-            x = op.concrete_forward(x)
+            x = op.concrete_forward(x)  # type: ignore
         return x
 
     def set_forward_mode(self, mode: str) -> None:

art/estimators/certification/interval/pytorch.py

@@ -179,7 +179,7 @@ def concrete_forward(self, in_x: np.ndarray | "torch.Tensor") -> "torch.Tensor":
             if isinstance(op, PyTorchIntervalConv2D) and self.forward_mode == "attack":
                 x = op.conv_forward(x)
             else:
-                x = op.concrete_forward(x)
+                x = op.concrete_forward(x)  # type: ignore
         return x
 
     def set_forward_mode(self, mode: str) -> None: