fix: Remove captum from project deps

dilyabareeva · Nov 21, 2024 · bf9ce8a · bf9ce8a
1 parent 7b60463
commit bf9ce8a
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -30,6 +30,9 @@ jobs:
       - name: Verify Python Installation
         run: python --version
 
+      - name: Install dependencies
+        run: pip install git+https://github.com/pytorch/captum.git
+
       - name: Install tox-gh
         run: python -m pip install tox-gh>=1.2
 

diff --git a/.github/workflows/type-lint.yml b/.github/workflows/type-lint.yml
@@ -17,6 +17,9 @@ jobs:
       - name: Install tox-gh
         run: pip install tox-gh
 
+      - name: Install dependencies
+        run: pip install git+https://github.com/pytorch/captum.git
+
       - name: Run flake8
         run: tox run -e lint
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -19,7 +19,6 @@ keywords = ["interpretability", "xai", "machine learning", "deep learning"]
 dependencies = [
     "numpy>=1.19.5",
     "torch>=1.13.1",
-    "captum@git+https://github.com/pytorch/captum",
     "lightning>=1.4.0",
     "torchmetrics>=1.4.0",
     "tqdm>=4.0.0",