From bf9ce8a01232538d5638ab52fa4e4071484ee1c0 Mon Sep 17 00:00:00 2001
From: dilyabareeva <dilyabareeva@gmail.com>
Date: Thu, 21 Nov 2024 18:22:46 +0100
Subject: [PATCH] fix: Remove captum from project deps

---
 .github/workflows/tests.yml     | 3 +++
 .github/workflows/type-lint.yml | 3 +++
 pyproject.toml                  | 1 -
 3 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 68ed4ade..94e78337 100644
--- 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
index 38f70f55..c6d5149b 100644
--- 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
index 37decb7f..4661c735 100644
--- 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",