From 2874d244066b64093eb6f130f2e418a1338f8299 Mon Sep 17 00:00:00 2001
From: Antonin RAFFIN <antonin.raffin@ensta.org>
Date: Thu, 17 Jan 2019 20:15:05 +0100
Subject: [PATCH] Update stable-baselines to 2.4.0

---
 docker/Dockerfile.gpu | 2 +-
 tests/test_enjoy.py   | 3 ---
 2 files changed, 1 insertion(+), 4 deletions(-)

diff --git a/docker/Dockerfile.gpu b/docker/Dockerfile.gpu
index 3caceea..db78d95 100644
--- a/docker/Dockerfile.gpu
+++ b/docker/Dockerfile.gpu
@@ -17,12 +17,12 @@ RUN \
     mkdir $CODE_DIR && \
     cd $CODE_DIR && \
     pip install --upgrade pip && \
-    pip install tensorflow-gpu==1.8.0 && \
     pip install pytest==3.5.1 && \
     pip install pytest-cov && \
     pip install pyyaml && \
     pip install box2d-py==2.3.5 && \
     pip install stable-baselines==2.4.0 && \
+    pip install tensorflow-gpu==1.8.0 && \
     pip install pybullet==2.1.0 && \
     pip install pytablewriter==0.36.0
 
diff --git a/tests/test_enjoy.py b/tests/test_enjoy.py
index 0b2e14c..7e7915e 100644
--- a/tests/test_enjoy.py
+++ b/tests/test_enjoy.py
@@ -18,9 +18,6 @@ def _assert_eq(left, right):
 @pytest.mark.parametrize("trained_model", trained_models.keys())
 def test_enjoy(trained_model):
     algo, env_id = trained_models[trained_model]
-    # Temp fix waiting for new stable-baselines version
-    if algo == "sac":
-        return
     args = [
         '-n', str(N_STEPS),
         '-f', FOLDER,