Merge pull request #117 from griffithlab/class_ii

Also fix DQ class II alleles
griffithlab · Apr 30, 2018 · 954e796 · 954e796
2 parents b4fae6a + 0a06eb2
commit 954e796
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/call_iedb.py b/lib/call_iedb.py
@@ -60,7 +60,7 @@ def main(args_input = sys.argv[1:]):
         data = {
             'sequence_text': args.input_file.read(),
             'method':        args.method,
-            'allele':        args.allele.replace('-DPB', '/DPB'),
+            'allele':        args.allele.replace('-DPB', '/DPB').replace('-DQB', '/DQB'),
             'user_tool':     'pVac-seq',
         }
         if args.epitope_length is not None:

diff --git a/lib/prediction_class.py b/lib/prediction_class.py
@@ -201,7 +201,7 @@ def iedb_executable_params(self, args):
             'python2.7',
             args.iedb_executable_path,
             args.method,
-            args.allele.replace('-DPB', '/DPB'),
+            args.allele.replace('-DPB', '/DPB').replace('-DQB', '/DQB'),
             args.input_file.name,
         ]