ModuleNotFoundError: No module named 'ochem_predict_nn'

[PengBo1118]

(my-rdkit-env) λ python data/generate_reaction_templates.py
Traceback (most recent call last):
File "data/generate_reaction_templates.py", line 27, in
from ochem_predict_nn.utils.database import collection_templates, collection_example_reactions_smilesonly
ModuleNotFoundError: No module named 'ochem_predict_nn'

[gnsrivastava]

ochem_predict_nn.utils.database will work if you are running the tool from outside the ochem_predict_nn. So if you are inside the ochem_predict_nn then use 'from utils.database import collection_templates, collection_example_reactions_smilesonly'