From 0ce2bf7cfb2b5447c3dc1b993bdf94bf09fa6994 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Thu, 22 Feb 2018 11:48:38 -0500
Subject: [PATCH] Update downloads to include model selected predictors

---
 mhcflurry/downloads.yml                 | 6 +++++-
 test/test_train_and_related_commands.py | 2 +-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/mhcflurry/downloads.yml b/mhcflurry/downloads.yml
index 4c01ebaa..0b07827b 100644
--- a/mhcflurry/downloads.yml
+++ b/mhcflurry/downloads.yml
@@ -21,7 +21,11 @@ releases:
         compatibility-version: 2
         downloads:
             - name: models_class1
-              url: http://github.com/openvax/mhcflurry/releases/download/pre-1.1/models_class1.20180205.tar.bz2
+              url: https://github.com/openvax/mhcflurry/releases/download/pre-1.2/models_class1.20180222.tar.bz2
+              default: true
+
+            - name: models_class1_selected_no_mass_spec
+              url: https://github.com/openvax/mhcflurry/releases/download/pre-1.2/models_class1_selected_no_mass_spec.20180222.tar.bz2
               default: true
 
             - name: models_class1_unselected
diff --git a/test/test_train_and_related_commands.py b/test/test_train_and_related_commands.py
index 0e8e3d76..ef5664af 100644
--- a/test/test_train_and_related_commands.py
+++ b/test/test_train_and_related_commands.py
@@ -134,7 +134,7 @@ def run_and_check_with_model_selection(n_jobs=1):
         "--out-models-dir", models_dir2,
         "--models-dir", models_dir1,
         "--num-jobs", str(n_jobs),
-        "--max-models", "1"
+        "--mse-max-models", "1"
     ]
     print("Running with args: %s" % args)
     subprocess.check_call(args)
-- 
GitLab