Main.py doesn't work because cnn has no attribute 'predict_with_module'

[gulseva]

Hi,

I tried to run the main.py script but I get the following error:

Traceback (most recent call last):
File "main.py", line 2, in
from prediction.prediction import run_predictions
File "/home/gulseva/programs/Ca-Backbone-Prediction/prediction/init.py", line 1, in
from . import prediction
File "/home/gulseva/programs/Ca-Backbone-Prediction/prediction/prediction.py", line 24, in
cnn.predict_with_module,
AttributeError: module 'cnn' has no attribute 'predict_with_module'

The latest change to these files was 8 days ago so I assume something went wrong at that point. I would appreciate if you could take a look.