Error running testing

[Sherry40931]

Hi, thank you very much for your work.

I have downloaded the pretrained model and tried to run the testing in the README: ./translate_one_image.py --config ../exps/celeba_blond_hair.yaml --image_name ../images/ian.jpg --output_image_name ../results/ian_to_eyeglasses.jpg --weights ../snapshots/celeba_eyeglasses_gen_00500000.pkl --a2b 0

And I face this error:

self.image_save_iterations=500
self.display=1
self.snapshot_prefix='../outputs/celeba_blond_hair/celeba_blond_hair'
self.hyperparameters={'gen': 'CoVAE', 'dis': 'CoDis', 'vae_enc_w': 1e-05, 'vae_ll_w': 0.0001, 'gan_w': 1.0, 'batch_size': 1, 'ch': 64, 'max_iterations': 500000}
self.datasets={'a': {'channels': 3, 'image_size': 128, 'scale': 0, 'class_name': 'dataset_image', 'root': '../datasets/celeba/', 'folder': 'img_align_crop_resize_celeba/', 'list': 'lists/Blond_Hair_ON.txt'}, 'b': {'channels': 3, 'image_size': 128, 'scale': 0, 'class_name': 'dataset_image', 'root': '../datasets/celeba/', 'folder': 'img_align_crop_resize_celeba/', 'list': 'lists/Blond_Hair_OFF.txt'}}
Traceback (most recent call last):
  File "./translate_one_image.py", line 116, in <module>
    main(sys.argv)
  File "./translate_one_image.py", line 86, in main
    trainer = unit_trainer.UNITTrainer(gen_net, dis_net, batch_size, ch, input_dims, image_size)
  File "/home/iis/Documents/UNIT/src/trainers/unit_trainer.py", line 52, in __init__
    exec( 'self.dis = %s(ch, true_input_dims)' % dis)
  File "<string>", line 1, in <module>
NameError: name 'CoDis' is not defined

Can you tell me what did I do wrong? Thank you beforehand.