paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-04-25 10:49:30 -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
# 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