]> git.ipfire.org Git - thirdparty/paperless-ngx.git/commitdiff
catch another exception regarding classifier loading
authorjonaswinkler <17569239+jonaswinkler@users.noreply.github.com>
Wed, 19 May 2021 20:57:52 +0000 (22:57 +0200)
committerjonaswinkler <17569239+jonaswinkler@users.noreply.github.com>
Wed, 19 May 2021 20:57:52 +0000 (22:57 +0200)
src/documents/classifier.py

index 664c0022241f1199846d19857b8718db82b41d96..0069781b566334dcc24b02ae57f2f1000bf725da 100644 (file)
@@ -34,7 +34,9 @@ def load_classifier():
     try:
         classifier.load()
 
-    except (EOFError, IncompatibleClassifierVersionError) as e:
+    except (EOFError,
+            IncompatibleClassifierVersionError,
+            pickle.UnpicklingError):
         # there's something wrong with the model file.
         logger.exception(
             f"Unrecoverable error while loading document "
@@ -42,9 +44,9 @@ def load_classifier():
         )
         os.unlink(settings.MODEL_FILE)
         classifier = None
-    except OSError as e:
-        logger.error(
-            f"Error while loading document classification model: {str(e)}"
+    except OSError:
+        logger.exception(
+            f"Error while loading document classification model"
         )
         classifier = None