modify setup.py
Merge pull request #1852 from dongweiming/modify_setup

modify setup.py