From 3580c4c899c6d5b2d1954b333f36c1242382ca22 Mon Sep 17 00:00:00 2001 From: Trenton Holmes <797416+stumpylog@users.noreply.github.com> Date: Sun, 19 Jan 2025 08:05:22 -0800 Subject: [PATCH] Attempt to resolve via ordering --- src/documents/classifier.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/src/documents/classifier.py b/src/documents/classifier.py index 866e03969..f576fc633 100644 --- a/src/documents/classifier.py +++ b/src/documents/classifier.py @@ -169,7 +169,7 @@ class DocumentClassifier: tags__is_inbox_tag=True, ) .select_related("document_type", "correspondent", "storage_path") - .prefetch_related("tags") + .prefetch_related("tags").order_by("pk") ) # No documents exit to train against @@ -199,12 +199,7 @@ class DocumentClassifier: hasher.update(y.to_bytes(4, "little", signed=True)) labels_correspondent.append(y) - tags: list[int] = sorted( - tag.pk - for tag in doc.tags.filter( - matching_algorithm=MatchingModel.MATCH_AUTO, - ) - ) + tags: list[int] = list(doc.tags.filter(matching_algorithm=MatchingModel.MATCH_AUTO).order_by("pk").values_list("pk", flat=True)) for tag in tags: hasher.update(tag.to_bytes(4, "little", signed=True)) labels_tags.append(tags)