Merge pull request #325 from griffithlab/netmhccons_error

Retry making prediction if NetMHCcons throws `len(peptide_list) != len(scores)` error

lib/call_iedb.py

@@ -45,7 +45,14 @@ def main(args_input = sys.argv[1:]):
 
     prediction_class = getattr(sys.modules[__name__], args.method)
     prediction_class_object = prediction_class()
-    (response_text, output_mode) = prediction_class_object.predict(args.input_file, args.allele, args.epitope_length, args.iedb_executable_path, args.iedb_retries)
+
+    try:
+        (response_text, output_mode) = prediction_class_object.predict(args.input_file, args.allele, args.epitope_length, args.iedb_executable_path, args.iedb_retries)
+    except Exception as err:
+        if str(err) == 'len(peptide_list) != len(scores)':
+            (response_text, output_mode) = prediction_class_object.predict(args.input_file, args.allele, args.epitope_length, args.iedb_executable_path, args.iedb_retries)
+        else:
+            raise err
 
     tmp_output_file = args.output_file + '.tmp'
     if output_mode == 'pandas':