From 45ec94c3340fd7040643f38d322c4608c8e887e9 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Tue, 20 Feb 2018 13:31:12 -0500
Subject: [PATCH] update downloads.yml
---
mhcflurry/downloads.yml | 4 ++++
mhcflurry/select_allele_specific_models_command.py | 4 ++--
2 files changed, 6 insertions(+), 2 deletions(-)
diff --git a/mhcflurry/downloads.yml b/mhcflurry/downloads.yml
index fc9d0416..a4661caa 100644
--- a/mhcflurry/downloads.yml
+++ b/mhcflurry/downloads.yml
@@ -28,6 +28,10 @@ releases:
url: https://github.com/openvax/mhcflurry/releases/download/pre-1.2/models_class1_unselected.20180216.tar.bz2
default: false
+ - name: models_class1_consensus
+ url: https://github.com/openvax/mhcflurry/releases/download/pre-1.2/models_class1_consensus.tar.bz2
+ default: false
+
- name: data_iedb
url: https://github.com/openvax/mhcflurry/releases/download/pre-1.0/data_iedb.tar.bz2
default: false
diff --git a/mhcflurry/select_allele_specific_models_command.py b/mhcflurry/select_allele_specific_models_command.py
index 18fb6e51..577c1d41 100644
--- a/mhcflurry/select_allele_specific_models_command.py
+++ b/mhcflurry/select_allele_specific_models_command.py
@@ -197,7 +197,7 @@ def run(argv=sys.argv[1:]):
print("Attempting to create directory: %s" % args.out_models_dir)
os.mkdir(args.out_models_dir)
print("Done.")
-
+
result_predictor = Class1AffinityPredictor(metadata_dataframes=metadata_dfs)
worker_pool = worker_pool_with_gpu_assignments_from_args(args)
@@ -230,7 +230,7 @@ def run(argv=sys.argv[1:]):
worker_pool.join()
print("Model selection time %0.2f min." % (model_selection_time / 60.0))
- print("Predictor written to: %s" % args.models_dir)
+ print("Predictor written to: %s" % args.out_models_dir)
def model_select(allele):
--
GitLab