helpy4/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich synced 2026-08-15 13:03:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

chore(ml-test): remove unused cache_dir parameter from TextRecognizer

Browse source
This commit is contained in:
pneuly 2026-05-26 12:12:37 +09:00
parent 803cc566f2
commit 007ad7b3f2
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
machine-learning/test_main.py
View file

@ -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(