Adding async future functionality to ShapleyValues (#1487)

Summary: This diff implements the attribute_future method for the ShapleyValueSampling class. Reviewed By: cyrjano Differential Revision: D68158802
pytorch · Jan 21, 2025 · 2ad525d · 2ad525d
1 parent b487891
commit 2ad525d
Show file tree

Hide file tree

Showing 4 changed files with 452 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