Can't run train_pred_polyinfo.py after following the README instruction

[WwwwNB]

Hi, I ran the following command:

git clone https://github.com/learningmatter-mit/Chem-prop-pred.git
cd Chem-prop-pred
git submodule init
git submodule update
cd ChemArr
conda env create -f environment.yml
conda activate chemprop
pip install -e .
cd ../
pip install -e .

And using conda list I can see all the packages in the env. But here I encounter 2 problems.

1. python train_pred_polyinfo.py --make_data true --train_predict true --polyinfo_datafiles true.

2. On vscode, import chemprop cannot be recognized even after reinstalling chemprop

Thanks,