mirror of
https://github.com/paperless-ngx/paperless-ngx.git
synced 2025-07-28 18:24:38 -05:00
Updates scikit-learn and fixes the warning this normally generates in the unit tests
This commit is contained in:

committed by
Johann Bauer

parent
00059e9482
commit
ba79aff89b
@@ -204,7 +204,12 @@ class TestClassifier(DirectoriesMixin, TestCase):
|
||||
MODEL_FILE=os.path.join(os.path.dirname(__file__), "data", "model.pickle"),
|
||||
)
|
||||
def test_load_and_classify(self):
|
||||
# Generate test data, train and save to the model file
|
||||
# This ensures the model file sklearn version matches
|
||||
# and eliminates a warning
|
||||
self.generate_test_data()
|
||||
self.classifier.train()
|
||||
self.classifier.save()
|
||||
|
||||
new_classifier = DocumentClassifier()
|
||||
new_classifier.load()
|
||||
|
Reference in New Issue
Block a user