Add vit attribution hugo

.pre-commit-config.yaml

@@ -24,7 +24,7 @@ repos:
         exclude: LICENSE
 
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.15.10
+    rev: v0.15.22
     hooks:
       - id: ruff-format
       - id: ruff

interpreto/__init__.py

@@ -29,6 +29,16 @@
 
 from .attributions import (
     GradientShap,
+    ImageGradientShap,
+    ImageIntegratedGradients,
+    ImageKernelShap,
+    ImageLime,
+    ImageOcclusion,
+    ImageSaliency,
+    ImageSmoothGrad,
+    ImageSobol,
+    ImageSquareGrad,
+    ImageVarGrad,
     IntegratedGradients,
     KernelShap,
     Lime,
@@ -40,7 +50,8 @@
     VarGrad,
 )
 from .commons import (
-    Granularity,
+    ImageGranularity,
+    TextGranularity,
 )
 from .model_wrapping import ModelWithSplitPoints
 from .visualizations import (
@@ -62,7 +73,18 @@ def get_version() -> str:
 
 __all__ = [
     "GradientShap",
-    "Granularity",
+    "TextGranularity",
+    "ImageGranularity",
+    "ImageGradientShap",
+    "ImageIntegratedGradients",
+    "ImageKernelShap",
+    "ImageLime",
+    "ImageOcclusion",
+    "ImageSaliency",
+    "ImageSmoothGrad",
+    "ImageSobol",
+    "ImageSquareGrad",
+    "ImageVarGrad",
     "IntegratedGradients",
     "KernelShap",
     "Lime",

interpreto/attributions/__init__.py

@@ -25,6 +25,16 @@
 from .base import InferenceModes
 from .methods import (
     GradientShap,
+    ImageGradientShap,
+    ImageIntegratedGradients,
+    ImageKernelShap,
+    ImageLime,
+    ImageOcclusion,
+    ImageSaliency,
+    ImageSmoothGrad,
+    ImageSobol,
+    ImageSquareGrad,
+    ImageVarGrad,
     IntegratedGradients,
     KernelShap,
     Lime,
@@ -38,6 +48,16 @@
 
 __all__ = [
     "GradientShap",
+    "ImageGradientShap",
+    "ImageIntegratedGradients",
+    "ImageKernelShap",
+    "ImageLime",
+    "ImageOcclusion",
+    "ImageSaliency",
+    "ImageSmoothGrad",
+    "ImageSobol",
+    "ImageSquareGrad",
+    "ImageVarGrad",
     "IntegratedGradients",
     "KernelShap",
     "Lime",

interpreto/attributions/aggregations/base.py

@@ -76,9 +76,9 @@ class TorchAggregator(Aggregator):
     @jaxtyped(typechecker=beartype)
     def aggregate(
         self,
-        results: Float[Tensor, "p t l"],
+        results: Float[Tensor, "p t l"] | Float[Tensor, "p t d l"],
         mask: torch.Tensor | None = None,
-    ) -> Float[Tensor, "t l"]:
+    ) -> Float[Tensor, "t l"] | Float[Tensor, "t d l"]:
         return self._method(results, dim=0)  # type: ignore
 
 
@@ -170,7 +170,7 @@ class TrapezoidalMeanAggregator(Aggregator):
     @jaxtyped(typechecker=beartype)
     def aggregate(
         self,
-        results: Float[Tensor, "p t l"],
+        results: Float[Tensor, "p t l"] | Float[Tensor, "p t d l"],
         mask: torch.Tensor | None = None,
-    ) -> Float[Tensor, "t l"]:
+    ) -> Float[Tensor, "t l"] | Float[Tensor, "t d l"]:
         return torch.trapezoid(results, dim=0) / (results.shape[0] - 1)