Update stable-baselines to 2.4.0

araffin · Jan 17, 2019 · 2874d24 · 2874d24
1 parent c6ce4ba
commit 2874d24
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git 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
@@ -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,