Fix get_config_util pyre fixme issues

Summary: Fixing unresolved pyre fixme issues in corresponding file

Differential Revision: D76737143

Author: Vivek Miglani

captum/testing/attr/helpers/get_config_util.py

@@ -1,7 +1,7 @@
 # (c) Meta Platforms, Inc. and affiliates. Confidential and proprietary.
 
 # pyre-strict
-from typing import Any, Tuple
+from typing import List, Tuple
 
 import torch
 from captum._utils.gradient import compute_gradients
@@ -10,17 +10,15 @@
 from torch.nn import Module
 
 
-# pyre-fixme[3]: Return annotation cannot contain `Any`.
-def get_basic_config() -> Tuple[Module, Tensor, Tensor, Any]:
+def get_basic_config() -> Tuple[Module, Tensor, Tensor, None]:
     input = torch.tensor([1.0, 2.0, 3.0, 0.0, -1.0, 7.0], requires_grad=True).T
     # manually percomputed gradients
     grads = torch.tensor([-0.0, -0.0, -0.0, 1.0, 1.0, -0.0])
     return BasicModel(), input, grads, None
 
 
-# pyre-fixme[3]: Return annotation cannot contain `Any`.
 def get_multiargs_basic_config() -> (
-    Tuple[Module, Tuple[Tensor, ...], Tuple[Tensor, ...], Any]
+    Tuple[Module, Tuple[Tensor, ...], Tuple[Tensor, ...], Tuple[List[int], int]]
 ):
     model = BasicModel5_MultiArgs()
     additional_forward_args = ([2, 3], 1)
@@ -34,9 +32,8 @@ def get_multiargs_basic_config() -> (
     return model, inputs, grads, additional_forward_args
 
 
-# pyre-fixme[3]: Return annotation cannot contain `Any`.
 def get_multiargs_basic_config_large() -> (
-    Tuple[Module, Tuple[Tensor, ...], Tuple[Tensor, ...], Any]
+    Tuple[Module, Tuple[Tensor, ...], Tuple[Tensor, ...], Tuple[List[int], int]]
 ):
     model = BasicModel5_MultiArgs()
     additional_forward_args = ([2, 3], 1)