From fc83dd0b53c37c04e92fe1fc44023bb3d572c0a4 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Sun, 28 Jan 2018 16:18:27 -0500
Subject: [PATCH] fix
---
mhcflurry/train_allele_specific_models_command.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/mhcflurry/train_allele_specific_models_command.py b/mhcflurry/train_allele_specific_models_command.py
index 8646a020..bf8d653c 100644
--- a/mhcflurry/train_allele_specific_models_command.py
+++ b/mhcflurry/train_allele_specific_models_command.py
@@ -275,7 +275,7 @@ def run(argv=sys.argv[1:]):
results = (
calibrate_percentile_ranks(
allele=allele,
- predictor=args.out_model_dir,
+ predictor=predictor,
peptides=encoded_peptides)
for allele in alleles)
else:
@@ -292,7 +292,7 @@ def run(argv=sys.argv[1:]):
results = worker_pool.imap_unordered(
partial(
calibrate_percentile_ranks,
- predictor=predictor),
+ predictor=args.out_models_dir),
alleles,
chunksize=1)
--
GitLab