Backupppert/immich
1
1
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2025-07-21 22:44:23 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions30

feat(ml): ML on Rockchip NPUs ()

Browse source
This commit is contained in:
Yoni Yang 2025-03-18 00:04:08 +08:00 • committed by GitHub
commit 14c3b99c0f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
43 changed files with 2417 additions and 4726 deletions
machine-learning/app
conftest.py

6
machine-learning/app/conftest.py
View file

@ -136,6 +136,12 @@ def ann_session() -> Iterator[mock.Mock]:
yield mocked
@pytest.fixture(scope="function")
def rknn_session() -> Iterator[mock.Mock]:
with mock.patch("app.sessions.rknn.RknnPoolExecutor") as mocked:
yield mocked
@pytest.fixture(scope="function")
def rmtree() -> Iterator[mock.Mock]:
with mock.patch("app.models.base.rmtree", autospec=True) as mocked: