Commit 15e30c1d authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): fix unittest

parent 6495a5c3

test/upscale/test_cdc.py

+9 −0
Original line number Diff line number Diff line
@@ -3,6 +3,15 @@ from PIL import Image 


from imgutils.metrics import psnr

from imgutils.upscale import upscale_with_cdc

from imgutils.upscale.cdc import _open_cdc_upscaler_model





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

def _release_model():

    try:

        yield

    finally:

        _open_cdc_upscaler_model.cache_clear()





@pytest.mark.unittest