From e0c651e452056cac26237eef201cbb28c9990ecc Mon Sep 17 00:00:00 2001
From: Patrick Kidger <33688385+patrick-kidger@users.noreply.github.com>
Date: Tue, 20 Jul 2021 08:12:53 +0100
Subject: [PATCH 1/2] tweaks for PyPI release

---
 README.md | 2 +-
 setup.py  | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 457c5bc..767a95f 100644
--- a/README.md
+++ b/README.md
@@ -8,7 +8,7 @@ This library provides [stochastic differential equation (SDE)](https://en.wikipe
 
 ## Installation
 ```shell script
-pip install git+https://github.com/google-research/torchsde.git
+pip install torchsde
 ```
 
 **Requirements:** Python >=3.6 and PyTorch >=1.6.0.
diff --git a/setup.py b/setup.py
index 48f3856..8cc8863 100644
--- a/setup.py
+++ b/setup.py
@@ -26,12 +26,16 @@
     else:
         raise RuntimeError("Unable to find __version__ string.")
 
+with open(os.path.join(here, 'README.md')) as f:
+    readme = f.read()
+
 setuptools.setup(
     name="torchsde",
     version=version,
     author="Xuechen Li",
     author_email="lxuechen@cs.toronto.edu",
     description="SDE solvers and stochastic adjoint sensitivity analysis in PyTorch.",
+    long_description=readme,
     url="https://github.com/google-research/torchsde",
     packages=setuptools.find_packages(exclude=['benchmarks', 'diagnostics', 'examples', 'tests']),
     install_requires=[

From c6cd91c270a5ae269a994aad7aa03e88ae82a6e1 Mon Sep 17 00:00:00 2001
From: Patrick Kidger <33688385+patrick-kidger@users.noreply.github.com>
Date: Wed, 21 Jul 2021 14:22:06 +0100
Subject: [PATCH 2/2] fix content type

---
 setup.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/setup.py b/setup.py
index 8cc8863..0ad627f 100644
--- a/setup.py
+++ b/setup.py
@@ -36,6 +36,7 @@
     author_email="lxuechen@cs.toronto.edu",
     description="SDE solvers and stochastic adjoint sensitivity analysis in PyTorch.",
     long_description=readme,
+    long_description_content_type="text/markdown",
     url="https://github.com/google-research/torchsde",
     packages=setuptools.find_packages(exclude=['benchmarks', 'diagnostics', 'examples', 'tests']),
     install_requires=[