I’ve tried cloning and running this along with all the other tricks I could find online with no luck:
#git clone https://github.com/pytorch/captum.git
#cd captum
#pip install -e .

git clone https://github.com/UserName/captum
cd captum
git checkout custom-branch
pip install -e .

If I use the above code, I can import captum but can’t import any submodules.
Any suggestions?