From 7cd4b7952a508754ca0f8e1bc9b6732d07dd9b21 Mon Sep 17 00:00:00 2001
From: Julia K <julia326@gmail.com>
Date: Wed, 8 Nov 2017 18:30:14 -0500
Subject: [PATCH] If output models directory doesn't exist, create it
---
.../train_allele_specific_models_command.py | 6 ++++++
1 file changed, 6 insertions(+)
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 92a68f14..6f0aa975 100644
--- a/mhcflurry/class1_affinity_prediction/train_allele_specific_models_command.py
+++ b/mhcflurry/class1_affinity_prediction/train_allele_specific_models_command.py
@@ -2,6 +2,7 @@
Train Class1 single allele models.
"""
+import os
import sys
import argparse
import json
@@ -93,6 +94,11 @@ def run(argv=sys.argv[1:]):
predictor = Class1AffinityPredictor()
+ if args.out_models_dir and not os.path.exists(args.out_models_dir):
+ print("Attempting to create directory: %s" % args.out_models_dir)
+ os.mkdir(args.out_models_dir)
+ print("Done.")
+
for (h, hyperparameters) in enumerate(hyperparameters_lst):
n_models = hyperparameters.pop("n_models")
--
GitLab