From 73afaf157c553d00c743d1bdb2ace51c9006d43d Mon Sep 17 00:00:00 2001
From: Antonin RAFFIN <antonin.raffin@ensta.org>
Date: Thu, 7 May 2020 12:19:29 +0200
Subject: [PATCH] Add version.txt to package

---
 docs/misc/changelog.rst       | 2 +-
 setup.py                      | 5 ++++-
 stable_baselines3/version.txt | 2 +-
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/docs/misc/changelog.rst b/docs/misc/changelog.rst
index c1c831f..4240d4a 100644
--- a/docs/misc/changelog.rst
+++ b/docs/misc/changelog.rst
@@ -3,7 +3,7 @@
 Changelog
 ==========
 
-Pre-Release 0.6.0a (WIP)
+Pre-Release 0.6.0a1 (WIP)
 ------------------------------
 
 
diff --git a/setup.py b/setup.py
index cd000f8..14fc11b 100644
--- a/setup.py
+++ b/setup.py
@@ -4,7 +4,7 @@ import subprocess
 from setuptools import setup, find_packages
 
 with open(os.path.join('stable_baselines3', 'version.txt'), 'r') as file_handler:
-    __version__ = file_handler.read()
+    __version__ = file_handler.read().strip()
 
 
 long_description = """
@@ -70,6 +70,9 @@ model = PPO('MlpPolicy', 'CartPole-v1').learn(10000)
 setup(name='stable_baselines3',
       packages=[package for package in find_packages()
                 if package.startswith('stable_baselines3')],
+      package_data={
+        'stable_baselines3': ['py.typed', 'version.txt']
+      },
       install_requires=[
           'gym[classic_control]>=0.11',
           'numpy',
diff --git a/stable_baselines3/version.txt b/stable_baselines3/version.txt
index b7c431c..18f5d16 100644
--- a/stable_baselines3/version.txt
+++ b/stable_baselines3/version.txt
@@ -1 +1 @@
-0.6.0a0
+0.6.0a1