fixed bug

main.py

@@ -6,10 +6,10 @@ from rdkit import RDLogger
 
 from grover.util.parsing import parse_args, get_newest_train_args
 from grover.util.utils import create_logger
-from grover.task.cross_validate import cross_validate
-from grover.task.fingerprint import generate_fingerprints
-from grover.task.predict import make_predictions, write_prediction
-from grover.task.pretrain import pretrain_model
+from task.cross_validate import cross_validate
+from task.fingerprint import generate_fingerprints
+from task.predict import make_predictions, write_prediction
+from task.pretrain import pretrain_model
 from grover.data.torchvocab import MolVocab
 
 

src/commands.py

@@ -25,6 +25,7 @@ def predict_from_csv(data_path, features_path, checkpoint_dir, output_path):
         f"--no_features_scaling "
         f"--output {output_path}"
     )
+    run(predict_cmd)
 
 
 def finetune(train_path, val_path, train_features_path, val_features_path,