From c1c423c7b25b21faa1c439661263091fff16b5e1 Mon Sep 17 00:00:00 2001
From: Trenton H <797416+stumpylog@users.noreply.github.com>
Date: Thu, 2 Apr 2026 13:48:32 -0700
Subject: [PATCH] Patches things more directly

---
 src/documents/tests/data/v1.17.4.model.pickle | Bin 714 -> 0 bytes
 src/documents/tests/test_classifier.py        |  29 ++++++++++++------
 2 files changed, 19 insertions(+), 10 deletions(-)
 delete mode 100644 src/documents/tests/data/v1.17.4.model.pickle

diff --git a/src/documents/tests/data/v1.17.4.model.pickle b/src/documents/tests/data/v1.17.4.model.pickle
deleted file mode 100644
index 4b2734607f8453ad7ecf7d972eea7eafb4deea4d..0000000000000000000000000000000000000000
GIT binary patch
literal 0
HcmV?d00001

literal 714
zcmZWn&ubGw6i(A5O`5b;(X$aeG!$dS(t7J5NRU7dJt#<Eob0}&OE)vi>}>Nxq!lz)
zc_%vxLiOa?lL!9<4<5vOp<YBUB3>2s&32&ycX`bB-hA)9Z{EyqrctQwW}e+j9e<d2
zxd!d^z)Zk>Q=6UnIyJv=rcyXsGMg#PTgpfZF#FamSl0z#bMhN0`#&-p<{{ab%yzh#
zbf}fbd!KH6`&fVb`|`VkquYO8y??s#_ru}ymtVgBymPqo>q1OFmrewo54U{aN>n?*
zb)<m7Kr8NOPl~E0Z17Y~Mnc~NM@!|6K#_#XeRIz&V%8HuM2s0n3Jro#9)u%ZPjn4t
zBi*@l&0xs|3Q=99q%xQbwelRI&RGyJ?l_bgnkJS41%Z+dgdq`|bZEH+BJSA4JQv&_
z+U}U?OXV6Y__7bm;USnkoQt_E;97$cRD5wgs<hgzeyhLKY8#vlwG5a|w4HK2K45Og
z)YpF`Q^@r^aU6Lt`doi=_AZtzCz*-ODT=P*J*K$mf@w@O(qpQ+EgjyDe6EJJYTFYQ
zJ;to7hXLJiRz$sENUJCG>I%&bPj|clxa=g45q+MWsAHaMp5)4}4_;i~!Qw>RN|;QG
z=f4TQFG&q%>1e)d8q+wvw!BJD1dH)rIf<vS5TkOun~89NCHKy@L4|bLr1DDjYIWH}
KZF)}4|Lh;_RT7c_

diff --git a/src/documents/tests/test_classifier.py b/src/documents/tests/test_classifier.py
index a1bb9baa5..5e8195166 100644
--- a/src/documents/tests/test_classifier.py
+++ b/src/documents/tests/test_classifier.py
@@ -1,4 +1,3 @@
-import pickle
 import re
 import warnings
 from pathlib import Path
@@ -447,18 +446,28 @@ class TestClassifier(DirectoriesMixin, TestCase):
 
         self.generate_train_and_save()
 
-        real_loads = pickle.loads
+        fake_warning = warnings.WarningMessage(
+            message=InconsistentVersionWarning(
+                estimator_name="MLPClassifier",
+                current_sklearn_version="1.0",
+                original_sklearn_version="0.9",
+            ),
+            category=InconsistentVersionWarning,
+            filename="",
+            lineno=0,
+        )
 
-        def loads_with_warning(data):
-            warnings.warn(
-                "Trying to unpickle estimator from version 0.0 when using version 1.0.",
-                InconsistentVersionWarning,
-            )
-            return real_loads(data)
+        real_catch_warnings = warnings.catch_warnings
+
+        class PatchedCatchWarnings(real_catch_warnings):
+            def __enter__(self):
+                w = super().__enter__()
+                w.append(fake_warning)
+                return w
 
         with mock.patch(
-            "documents.classifier.pickle.loads",
-            side_effect=loads_with_warning,
+            "documents.classifier.warnings.catch_warnings",
+            PatchedCatchWarnings,
         ):
             with self.assertRaises(IncompatibleClassifierVersionError):
                 self.classifier.load()