Fix kwargs annotation in feature_ablation.py

Summary: Fix incorrect **kwargs annotation

Differential Revision: D65001879

Author: craymichael

captum/attr/_core/feature_ablation.py

@@ -3,7 +3,7 @@
 # pyre-strict
 
 import math
-from typing import Any, Callable, cast, Dict, Generator, List, Optional, Tuple, Union
+from typing import Any, Callable, cast, Generator, List, Optional, Tuple, Union
 
 import torch
 from captum._utils.common import (
@@ -573,7 +573,7 @@ def _attribute_progress_setup(
         formatted_inputs: Tuple[Tensor, ...],
         feature_mask: Tuple[Tensor, ...],
         perturbations_per_eval: int,
-        **kwargs: Dict[str, Any],
+        **kwargs: Any,
     ):
         feature_counts = self._get_feature_counts(
             formatted_inputs, feature_mask, **kwargs