From 9d69edfada5b2c45820419b26e7240b5241219ae Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Sun, 28 Jan 2018 15:51:10 -0500
Subject: [PATCH] attempt fix deadlock

---
 mhcflurry/train_allele_specific_models_command.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mhcflurry/train_allele_specific_models_command.py b/mhcflurry/train_allele_specific_models_command.py
index c2b6ed5d..ce6231ab 100644
--- a/mhcflurry/train_allele_specific_models_command.py
+++ b/mhcflurry/train_allele_specific_models_command.py
@@ -252,6 +252,7 @@ def run(argv=sys.argv[1:]):
     if worker_pool:
         worker_pool.close()
         worker_pool.join()
+        worker_pool = None
 
     start = time.time()
     if args.percent_rank_calibration_num_peptides_per_length > 0:
@@ -305,6 +306,7 @@ def run(argv=sys.argv[1:]):
     if worker_pool:
         worker_pool.close()
         worker_pool.join()
+        worker_pool = None
 
     print("Train time: %0.2f min. Percent rank calibration time: %0.2f min." % (
         training_time / 60.0, percent_rank_calibration_time / 60.0))
-- 
GitLab