A little mistake in utils.py

davide97l / rl-policies-attacks-defenses

Adversarial attacks on Deep Reinforcement Learning (RL)

MIT License

76 stars 12 forks source link

A little mistake in utils.py #18

Closed neucc1997 closed 3 years ago

neucc1997 commented 3 years ago

In utils.py line 96. The code "elif args.image_attack == ['fgm', 'fgsm']:" will always have a wrong result since args.image_attack is a str type attribute. It should be modified as "elif args.image_attack in ['fgm', 'fgsm']:"

davide97l commented 3 years ago

Thank you very much for your observation, The mistake has already been corrected but not yet committed to the master branch, anyway I will fix it immediately also on master.