Adopt tests to bzip2'd ttyrecs.

facebookresearch · Oct 21, 2020 · db18df0 · db18df0
1 parent 946e983
commit db18df0
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/.circleci/config.yml b/.circleci/config.yml
@@ -54,7 +54,7 @@ commands:
           command: |
             sudo apt-get install software-properties-common
             sudo apt-get update -y
-            sudo apt-get install -y libncurses5 libncurses5-dev flex bison cmake
+            sudo apt-get install -y libncurses5 libncurses5-dev flex bison cmake libbz2-dev
 
   pip-install-nle:
     description: Installs NLE and its dependencies

diff --git a/README.md b/README.md
@@ -44,7 +44,7 @@ can be installed by doing:
 ```bash
 # Python and most build deps
 $ sudo apt-get install -y build-essential autoconf libtool pkg-config \
-    python3-dev python3-pip python3-numpy git libncurses5-dev flex bison
+    python3-dev python3-pip python3-numpy git libncurses5-dev flex bison libbz2-dev
 
 # recent cmake version
 $ wget -O - https://apt.kitware.com/keys/kitware-archive-latest.asc 2>/dev/null | sudo apt-key add -

diff --git a/nle/tests/test_envs.py b/nle/tests/test_envs.py
@@ -224,7 +224,7 @@ def test_rollout(self, env_name, rollout_len):
             env.close()
 
             assert os.path.exists(
-                os.path.join(savedir, "nle.%i.0.ttyrec" % os.getpid())
+                os.path.join(savedir, "nle.%i.0.ttyrec.bz2" % os.getpid())
             )
 
     def test_rollout_no_archive(self, env_name, rollout_len):