From 55dadea98ed2bcb0c74da8c271cd681304def0eb Mon Sep 17 00:00:00 2001 From: Trenton Holmes Date: Thu, 2 Jun 2022 14:41:49 -0700 Subject: [PATCH] No need for a branch here, the loop takes care of it --- src/documents/classifier.py | 21 ++++++++++----------- 1 file changed, 10 insertions(+), 11 deletions(-) diff --git a/src/documents/classifier.py b/src/documents/classifier.py index 1c2ccea07..594973ac6 100644 --- a/src/documents/classifier.py +++ b/src/documents/classifier.py @@ -106,17 +106,16 @@ class DocumentClassifier: # Check for the warning about unpickling from differing versions # and consider it incompatible - if len(w) > 0: - sk_learn_warning_url = ( - "https://scikit-learn.org/stable/" - "model_persistence.html" - "#security-maintainability-limitations" - ) - for warning in w: - if issubclass(warning.category, UserWarning): - w_msg = str(warning.message) - if sk_learn_warning_url in w_msg: - raise IncompatibleClassifierVersionError() + sk_learn_warning_url = ( + "https://scikit-learn.org/stable/" + "model_persistence.html" + "#security-maintainability-limitations" + ) + for warning in w: + if issubclass(warning.category, UserWarning): + w_msg = str(warning.message) + if sk_learn_warning_url in w_msg: + raise IncompatibleClassifierVersionError() def save(self): target_file = settings.MODEL_FILE