def parse_dataset()

in contactopt/arguments.py [0:0]

• 12 lines of code
• 5 McCabe index (conditional complexity)

def parse_dataset(args):
    """ Converts the --split argument into a dataset file """
    if args.split == 'aug':
        args.train_dataset = 'data/perturbed_contactpose_train.pkl'
        args.test_dataset = 'data/perturbed_contactpose_test.pkl'
    elif args.split == 'fine':
        args.test_dataset = 'data/contactpose_test.pkl'
    elif args.split == 'im':
        args.test_dataset = 'data/ho3d_image.pkl'
    elif args.split == 'demo':
        args.test_dataset = 'data/ho3d_image_demo.pkl'
    else:
        raise ValueError('Unknown dataset')