Fix

Author: vivekmig

scripts/install_via_pip.sh

@@ -5,15 +5,13 @@ set -e
 PYTORCH_NIGHTLY=false
 DEPLOY=false
 CHOSEN_TORCH_VERSION=-1
-INSTALL_MODE=test
 
 while getopts 'ndfv:' flag; do
   case "${flag}" in
     n) PYTORCH_NIGHTLY=true ;;
     d) DEPLOY=true ;;
     f) FRAMEWORKS=true ;;
     v) CHOSEN_TORCH_VERSION=${OPTARG};;
-    m) INSTALL_MODE=${OPTARG};;
     *) echo "usage: $0 [-n] [-d] [-f] [-v version] [-m install_mode]" >&2
        exit 1 ;;
     esac
@@ -40,7 +38,7 @@ export TERM=xterm
 pip install --upgrade pip --progress-bar off
 
 # install captum with dev deps
-pip install -e .[test] --progress-bar off
+pip install -e .[dev] --progress-bar off
 BUILD_INSIGHTS=1 python setup.py develop
 
 # install other frameworks if asked for and make sure this is before pytorch

setup.py

@@ -67,7 +67,7 @@ def report(*args):
     INSIGHTS_REQUIRES
     + TEST_REQUIRES
     + [
-        "black==24.2.0",
+        "black",
         "flake8",
         "sphinx",
         "sphinx-autodoc-typehints",