Only use load compat if we can't load the classifier

2017-05-24 13:29:59 -07:00
parent f5f7264077
commit 4acf05cf28
1 changed files with 7 additions and 4 deletions
--- 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. """
+    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):