Don't import in setup.py

Some dependencies may not be installed yet