diff --git a/src/documents/management/commands/document_create_classifier.py b/src/documents/management/commands/document_create_classifier.py index eb642d8a7..b662195a7 100644 --- a/src/documents/management/commands/document_create_classifier.py +++ b/src/documents/management/commands/document_create_classifier.py @@ -17,7 +17,10 @@ class Command(PaperlessCommand): def handle(self, *args, **options) -> None: start = time.monotonic() - with self.buffered_logging("paperless.tasks"): + with ( + self.buffered_logging("paperless.tasks"), + self.buffered_logging("paperless.classifier"), + ): train_classifier( scheduled=False, status_callback=lambda msg: self.console.print(f" {msg}"),