add captum to pyproject.toml

dilyabareeva · May 14, 2024 · e2585f3 · e2585f3
1 parent 4ec5063
commit e2585f3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -9,6 +9,7 @@ keywords = ["explainable ai", "xai", "machine learning", "deep learning"]
 dependencies = [
     "numpy>=1.19.5",
     "torch>=1.13.1",
+    "captum>=0.6.0",
 ]
 dynamic = ["version"]