From 51573830721cbf734e39e01f0f7358f933e11378 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Sat, 10 Feb 2018 13:50:10 -0500
Subject: [PATCH] fix
---
mhcflurry/train_allele_specific_models_command.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mhcflurry/train_allele_specific_models_command.py b/mhcflurry/train_allele_specific_models_command.py
index 1711a8a2..91dcedfc 100644
--- a/mhcflurry/train_allele_specific_models_command.py
+++ b/mhcflurry/train_allele_specific_models_command.py
@@ -309,7 +309,7 @@ def run(argv=sys.argv[1:]):
# as it goes so no saving is required at the end.
for _ in tqdm.trange(len(work_items)):
item = work_items.pop(0) # want to keep freeing up memory
- (_, work_predictor) = train_model_entrypoint(item)
+ work_predictor = train_model_entrypoint(item)
assert work_predictor is predictor
assert not work_items
--
GitLab