From 007ad7b3f22b0c93cfc237509b460eb6d5c51075 Mon Sep 17 00:00:00 2001 From: pneuly Date: Tue, 26 May 2026 12:12:37 +0900 Subject: [PATCH] chore(ml-test): remove unused `cache_dir` parameter from `TextRecognizer` --- machine-learning/test_main.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/machine-learning/test_main.py b/machine-learning/test_main.py index 1efd7b1722..e2c9fdd900 100644 --- a/machine-learning/test_main.py +++ b/machine-learning/test_main.py @@ -909,7 +909,7 @@ class TestOcr: def test_set_rec_min_score(self, path: mock.Mock) -> None: path.return_value.__truediv__.return_value.__truediv__.return_value.suffix = ".onnx" - text_recognizer = TextRecognizer("PP-OCRv5_mobile", min_score=0.8, cache_dir="test_cache") + text_recognizer = TextRecognizer("PP-OCRv5_mobile", min_score=0.8) assert text_recognizer.min_score == 0.8 @@ -920,7 +920,7 @@ class TestOcr: mocker.patch("immich_ml.models.base.InferenceModel.download") rapid_recognizer = mocker.patch("immich_ml.models.ocr.recognition.RapidTextRecognizer") - text_recognizer = TextRecognizer("PP-OCRv5_mobile", cache_dir="test_cache") + text_recognizer = TextRecognizer("PP-OCRv5_mobile") text_recognizer.load() rapid_recognizer.assert_called_once_with( @@ -938,7 +938,7 @@ class TestOcr: rapid_recognizer = mocker.patch("immich_ml.models.ocr.recognition.RapidTextRecognizer") mocker.patch.object(settings, "max_batch_size", MaxBatchSize(ocr=4)) - text_recognizer = TextRecognizer("PP-OCRv5_mobile", cache_dir="test_cache") + text_recognizer = TextRecognizer("PP-OCRv5_mobile") text_recognizer.load() rapid_recognizer.assert_called_once_with( @@ -958,7 +958,7 @@ class TestOcr: rapid_recognizer = mocker.patch("immich_ml.models.ocr.recognition.RapidTextRecognizer") mocker.patch.object(settings, "max_batch_size", MaxBatchSize(facial_recognition=3)) - text_recognizer = TextRecognizer("PP-OCRv5_mobile", cache_dir="test_cache") + text_recognizer = TextRecognizer("PP-OCRv5_mobile") text_recognizer.load() rapid_recognizer.assert_called_once_with(