add est_mol task

run_interaction.py

@@ -119,7 +119,7 @@ def main(args):
     if args.task in ['train', 'train_z_1', 'train_z_10', 'train_z_100', 'train_mol']:
         train(args, model, dataset)
 
-    elif args.task in ['test']:
+    elif args.task in ['test', 'test_mol']:
         test(args, model, dataset)