From 5e1cd4f6527e9cdb4204acf25fbe367e37364990 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Sun, 28 Jan 2018 15:59:51 -0500
Subject: [PATCH] attempt fix deadlock
---
mhcflurry/train_allele_specific_models_command.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/mhcflurry/train_allele_specific_models_command.py b/mhcflurry/train_allele_specific_models_command.py
index ce6231ab..b09194aa 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=predictor,
+ predictor=args.out_model_dir,
peptides=encoded_peptides)
for allele in alleles)
else:
@@ -383,6 +383,8 @@ def calibrate_percentile_ranks(allele, predictor, peptides=None):
global GLOBAL_DATA
if peptides is None:
peptides = GLOBAL_DATA["calibration_peptides"]
+ if isinstance(predictor, str):
+ predictor = Class1AffinityPredictor.load(predictor)
predictor.calibrate_percentile_ranks(
peptides=peptides,
alleles=[allele])
--
GitLab