paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-04-02 13:45:10 -05:00
Code Issues Packages Projects Releases Wiki Activity

No need for a branch here, the loop takes care of it

Browse Source
This commit is contained in:
Trenton Holmes 2022-06-02 14:41:49 -07:00 committed by Johann Bauer
parent 77fbbe95ff
commit 55dadea98e
1 changed files with 10 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/documents/classifier.py
View File

@ -106,17 +106,16 @@ class DocumentClassifier:
# Check for the warning about unpickling from differing versions # Check for the warning about unpickling from differing versions
# and consider it incompatible # and consider it incompatible
if len(w) > 0: sk_learn_warning_url = (
sk_learn_warning_url = ( "https://scikit-learn.org/stable/"
"https://scikit-learn.org/stable/" "model_persistence.html"
"model_persistence.html" "#security-maintainability-limitations"
"#security-maintainability-limitations" )
) for warning in w:
for warning in w: if issubclass(warning.category, UserWarning):
if issubclass(warning.category, UserWarning): w_msg = str(warning.message)
w_msg = str(warning.message) if sk_learn_warning_url in w_msg:
if sk_learn_warning_url in w_msg: raise IncompatibleClassifierVersionError()
raise IncompatibleClassifierVersionError()
def save(self): def save(self):
target_file = settings.MODEL_FILE target_file = settings.MODEL_FILE