Update circuits.py

Author: ProGamerGov

captum/optim/_utils/circuits.py

@@ -112,7 +112,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
             max2avg_pool2d(child)
 
 
-def ignore_layer(model, layer: torch.nn.Module) -> None:
+def ignore_layer(model, layer) -> None:
     """
     Replace target layers with layers that do nothing.
     This is useful for removing the nonlinear ReLU