Update setup.py

Change cosmo dep'