Add augment and relation packages to setup.py.