From 17c2dabc7ffa46e05c3b45c472b9da09e8e3c475 Mon Sep 17 00:00:00 2001
From: Antonin RAFFIN <antonin.raffin@ensta.org>
Date: Fri, 25 Sep 2020 12:50:52 +0200
Subject: [PATCH] Update CI

---
 .github/workflows/ci.yml | 1 +
 setup.py                 | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 1342908..62770eb 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -32,6 +32,7 @@ jobs:
         pip install stable_baselines3[extra,tests,docs]
         # Install master version
         pip install git+https://github.com/DLR-RM/stable-baselines3
+        pip install .
         # Use headless version
         pip install opencv-python-headless
     # - name: Build the doc
diff --git a/setup.py b/setup.py
index b7f6f3e..f03afd8 100644
--- a/setup.py
+++ b/setup.py
@@ -24,6 +24,8 @@ setup(
     package_data={"sb3_contrib": ["py.typed", "version.txt"]},
     install_requires=[
         "stable_baselines3[tests,docs]",
+        # For progress bar when using CRR
+        "tqdm"
         # Enable CMA
         # "cma",
     ],