From cbaa17b58709718d20e6e0f7aba71aa1988aa4be Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Sun, 28 Jan 2018 13:25:35 -0500
Subject: [PATCH] more logging
---
mhcflurry/train_allele_specific_models_command.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/mhcflurry/train_allele_specific_models_command.py b/mhcflurry/train_allele_specific_models_command.py
index 4f48bb62..138dcdba 100644
--- a/mhcflurry/train_allele_specific_models_command.py
+++ b/mhcflurry/train_allele_specific_models_command.py
@@ -171,6 +171,7 @@ def run(argv=sys.argv[1:]):
predictor = Class1AffinityPredictor()
if args.train_num_jobs == 1:
# Serial run
+ print("Running in serial.")
worker_pool = None
else:
worker_pool = Pool(
@@ -267,6 +268,7 @@ def run(argv=sys.argv[1:]):
if args.calibration_num_jobs == 1:
# Serial run
+ print("Running in serial.")
worker_pool = None
results = (
calibrate_percentile_ranks(
@@ -287,7 +289,9 @@ def run(argv=sys.argv[1:]):
results = worker_pool.imap_unordered(
partial(
calibrate_percentile_ranks,
- predictor=predictor), alleles, chunksize=1)
+ predictor=predictor),
+ alleles,
+ chunksize=1)
for result in tqdm.tqdm(results, ascii=True, total=len(alleles)):
predictor.allele_to_percent_rank_transform.update(result)
--
GitLab