Adding async future functionality to ShapleyValues (#1487)

Summary: Pull Request resolved: #1487 This diff implements the attribute_future method for the ShapleyValueSampling class. Reviewed By: cyrjano Differential Revision: D68158802 fbshipit-source-id: e683657677faf59a0c711bde6c6cce4af45e1533
pytorch · Jan 22, 2025 · efe7908 · efe7908
1 parent a255953
commit efe7908
Show file tree

Hide file tree

Showing 4 changed files with 467 additions and 17 deletions.
diff --git a/captum/_utils/exceptions.py b/captum/_utils/exceptions.py
@@ -9,3 +9,11 @@ class FeatureAblationFutureError(Exception):
     FeatureAblation attribution call"""
 
     pass
+
+
+class ShapleyValueFutureError(Exception):
+    """This custom error is raised when an error
+    occurs within the callback chain of a
+    ShapleyValue attribution call"""
+
+    pass