Add wrapper function more tests and README

openvax · Feb 21, 2016 · 2049b84 · 2049b84
1 parent 076921c
commit 2049b84
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -30,6 +30,7 @@ strongest_predicted_binder = epitope_collection[0]
 The following models are available in `mhctools`:
 * `NetMHC3`: requires locally installed version of [NetMHC 3.x](http://www.cbs.dtu.dk/services/NetMHC-3.4/)
 * `NetMHC4`: requires locally installed version of [NetMHC 4.x](http://www.cbs.dtu.dk/services/NetMHC/)
+* `NetMHC`: a wrapper function to automatically use `NetMHC3` or `NetMHC4` depending on what's installed.
 * `NetMHCpan`: requires locally installed version of [NetMHCpan](http://www.cbs.dtu.dk/services/NetMHCpan/)
 * `NetMHCIIpan`: requires locally installed version of [NetMHCIIpan](http://www.cbs.dtu.dk/services/NetMHCIIpan/)
 * `NetMHCcons`: requires locally installed version of [NetMHCcons](http://www.cbs.dtu.dk/services/NetMHCcons/)

diff --git a/test/test_known_class1_epitopes.py b/test/test_known_class1_epitopes.py
@@ -21,6 +21,7 @@
     mhctools.NetMHCcons,
     mhctools.NetMHCpan,
     mhctools.NetMHCcons,
+    mhctools.NetMHC,
     mhctools.NetMHC3,
     mhctools.NetMHC4,
     mhctools.IedbNetMHCcons,