From 4acf05cf2875e31e3e63db99a269ffca73c85ccd Mon Sep 17 00:00:00 2001 From: Yacine Filali Date: Wed, 24 May 2017 13:29:59 -0700 Subject: [PATCH] Only use load compat if we can't load the classifier --- talon/signature/learning/classifier.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/talon/signature/learning/classifier.py b/talon/signature/learning/classifier.py index 4e1e886..0519713 100644 --- a/talon/signature/learning/classifier.py +++ b/talon/signature/learning/classifier.py @@ -30,11 +30,14 @@ def train(classifier, train_data_filename, save_classifier_filename=None): def load(saved_classifier_filename, train_data_filename): """Loads saved classifier. """ - import sys - if sys.version_info > (3, 0): - return load_compat(saved_classifier_filename) + try: + return joblib.load(saved_classifier_filename) + except Exception: + import sys + if sys.version_info > (3, 0): + return load_compat(saved_classifier_filename) - return joblib.load(saved_classifier_filename) + raise def load_compat(saved_classifier_filename):