Commit cc28c6cf authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): release memory after tagging models are tested

parent e1f712e4

test/tagging/test_deepdanbooru.py

+2 −2
Original line number Diff line number Diff line
@@ -5,7 +5,7 @@ from imgutils.tagging.deepdanbooru import _get_deepdanbooru_model 
from test.testings import get_testfile





@pytest.fixture()

@pytest.fixture(autouse=True, scope='module')

def _release_model_after_run():

    try:

        yield
@@ -15,7 +15,7 @@ def _release_model_after_run(): 


@pytest.mark.unittest

class TestTaggingDeepdanbooru:

    def test_get_deepdanbooru_tags(self, _release_model_after_run):

    def test_get_deepdanbooru_tags(self):

        rating, tags, chars = get_deepdanbooru_tags(get_testfile('6124220.jpg'))

        assert rating['rating:safe'] > 0.9

        assert tags['greyscale'] >= 0.8

test/tagging/test_mldanbooru.py

+1 −1
Original line number Diff line number Diff line
@@ -5,7 +5,7 @@ from imgutils.tagging.mldanbooru import _open_mldanbooru_model 
from test.testings import get_testfile





@pytest.fixture()

@pytest.fixture(autouse=True, scope='module')

def _release_model_after_run():

    try:

        yield

test/tagging/test_wd14.py

+1 −1
Original line number Diff line number Diff line
@@ -5,7 +5,7 @@ from imgutils.tagging.wd14 import _get_wd14_model 
from test.testings import get_testfile





@pytest.fixture()

@pytest.fixture(autouse=True, scope='module')

def _release_model_after_run():

    try:

        yield