| """ | |
| Shim for `captum.attr` exposing `IntegratedGradients` to match the real API used in the repo. | |
| """ | |
| from typing import Any, Tuple | |
| class IntegratedGradients: | |
| def __init__(self, model: Any): | |
| self.model = model | |
| def attribute(self, inputs: Any, target: int = 0, return_convergence_delta: bool = False) -> Tuple[Any, Any]: | |
| import numpy as np | |
| attr = np.zeros_like(inputs) | |
| delta = 0.0 | |
| return attr, delta | |
| __all__ = ["IntegratedGradients"] | |