Commit 52f61279 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): fix bug in unittest

parent c77de73b

test/metrics/test_lpips.py

+5 −2
Original line number Diff line number Diff line
@@ -15,8 +15,11 @@ from ..testings import get_testfile 
@keep_global_state()

def sampling_from_dataset(seed):

    global_seed(seed)

    files = sorted([file for file in glob.glob(os.path.join(get_testfile('dataset'), '**', '*'), recursive=True)

                    if os.path.isfile(file) and file.endswith('.jpg')])

    files = sorted([

        file for file in

        glob.glob(os.path.join(get_testfile('dataset'), 'chafen_arknights', '**', '*'), recursive=True)

        if os.path.isfile(file) and file.endswith('.jpg')

    ])



    sample = random.sample(files, 50)

    t_exists = {}