From b651138e25b5ef239cf0ace15cb07d708afaeb62 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Wed, 17 May 2017 17:32:50 -0400
Subject: [PATCH] Configure logging

---
 .../train_allele_specific_models_command.py                   | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/mhcflurry/class1_affinity_prediction/train_allele_specific_models_command.py b/mhcflurry/class1_affinity_prediction/train_allele_specific_models_command.py
index fd5172a5..e8a67737 100644
--- a/mhcflurry/class1_affinity_prediction/train_allele_specific_models_command.py
+++ b/mhcflurry/class1_affinity_prediction/train_allele_specific_models_command.py
@@ -14,7 +14,7 @@ import mhcnames
 
 
 from .class1_binding_predictor import Class1BindingPredictor
-
+from ..common import configure_logging
 
 def normalize_allele_name(s):
     try:
@@ -65,6 +65,8 @@ parser.add_argument(
 def run():
     args = parser.parse_args(sys.argv[1:])
 
+    configure_logging(verbose=args.verbosity > 1)
+
     hyperparameters_lst = json.load(open(args.hyperparameters))
     assert isinstance(hyperparameters_lst, list)
     print("Loaded hyperparameters list: %s" % str(hyperparameters_lst))
-- 
GitLab