Commit a7ea6054 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): fix bug in unittest

parent e4c98a5e
Loading
Loading
Loading
Loading
+4 −3
Original line number Diff line number Diff line
@@ -3,8 +3,9 @@ import os.path

import pytest

from imgutils.validate import anime_real
from imgutils.validate.real import _open_anime_real_model, anime_real_score
from imgutils.generic.classify import _open_models_for_repo_id
from imgutils.validate import anime_real, anime_real_score
from imgutils.validate.real import _REPO_ID
from test.testings import get_testfile

_ROOT_DIR = get_testfile('real')
@@ -19,7 +20,7 @@ def _release_model_after_run():
    try:
        yield
    finally:
        _open_anime_real_model.cache_clear()
        _open_models_for_repo_id(_REPO_ID).clear()


@pytest.mark.unittest