Fix setup.py to work with 'pip install -e .'
Merge pull request #4874 from leth/editable_install

Fix setup.py to work with 'pip install -e .'