from .base_options import BaseOptions


class TestOptions(BaseOptions):
    def initialize(self, parser):
        parser = BaseOptions.initialize(self, parser)
        parser.add_argument('--model_path')
        parser.add_argument('--no_resize', action='store_true')
        parser.add_argument('--no_crop', action='store_true')
        parser.add_argument('--eval', action='store_true', help='use eval mode during test time.')
                 
        parser.add_argument('--real_list_path', default='/mnt/data2/group2024-lhj/t2v/data/test/true', help='path for the list of real video,')
        parser.add_argument('--fake_list_path', default='/mnt/data2/group2024-lhj/t2v/data/test/fake', help='path for the list of fake video,')

        parser.add_argument('--ckpt', type=str, default='./checkpoints/best_network.pth')
        parser.add_argument('--output', type=str, default='./checkpoints/test')

        self.isTrain = False
        return parser