From ff8ccfff9c3a46436066af6fc5bb4c3973629b9d Mon Sep 17 00:00:00 2001
From: Francois <francois@orioncloud.fr>
Date: Sun, 10 May 2020 19:56:12 +0200
Subject: [PATCH] Adds link to the readme

---
 README.md |  2 +-
 setup.py  | 30 +++++++++++++++++++++++++-----
 2 files changed, 26 insertions(+), 6 deletions(-)

diff --git a/README.md b/README.md
index 8b19c4c..c83b728 100644
--- a/README.md
+++ b/README.md
@@ -34,7 +34,7 @@ F = - jax.hessian(likelihood)(cosmo)
 This is how you can compute gradients and hessians of any functions in `jax-cosmo`,
 all of this without any finite differences.
 
-Check out a full example here:
+Check out a full example here: [![colab link](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/DifferentiableUniverseInitiative/jax_cosmo/blob/master/notebooks/jax-cosmo-intro.ipynb)
 
 ## What is JAX?
 
diff --git a/setup.py b/setup.py
index 582c536..c187f38 100644
--- a/setup.py
+++ b/setup.py
@@ -1,11 +1,31 @@
 #!/usr/bin/env python
+import io
+import os
 from setuptools import setup, find_packages
 
-setup(name='jax_cosmo',
-      version='0.0.1',
+this_directory = os.path.abspath(os.path.dirname(__file__))
+with io.open(os.path.join(this_directory, 'README.md'), encoding='utf-8') as f:
+  long_description = f.read()
+
+setup(name='jax-cosmo',
       description='Differentiable Python Cosmology Library',
-      author='JAX Cosmo developers',
+      long_description=long_description,
+      long_description_content_type='text/markdown',
+      author='jax-cosmo developers',
       packages=find_packages(),
+      url='https://github.com/DifferentiableUniverseInitiative/jax_cosmo',
       install_requires=['jax', 'jaxlib'],
-      tests_require = ['pyccl']
-     )
+      tests_require = ['pyccl'],
+      use_scm_version=True,
+      setup_requires=['setuptools_scm'],
+      classifiers=[
+          'Programming Language :: Python :: 3.6',
+          'Programming Language :: Python :: 3.7',
+          'Programming Language :: Python :: 3.8',
+          'License :: OSI Approved :: MIT License',
+          'Operating System :: MacOS',
+          'Operating System :: POSIX :: Linux',
+          'Topic :: Scientific/Engineering',
+          'Intended Audience :: Science/Research',
+          'Intended Audience :: Developers',
+      ])