From 1f0ef53ceb63f5532df3eef53d8eda163dc354dd Mon Sep 17 00:00:00 2001
From: Andrii Tytarenko <titarenkoan@gmail.com>
Date: Thu, 2 May 2019 17:59:00 +0300
Subject: [PATCH] Fixed render function in vector environment (#71)
* Fixed render function in vector environment
* Update test_habitat_env.py
* Add message to assert
---
habitat/core/vector_env.py | 2 +-
test/test_habitat_env.py | 1 +
2 files changed, 2 insertions(+), 1 deletion(-)
diff --git a/habitat/core/vector_env.py b/habitat/core/vector_env.py
index 124918264..67c405a4d 100644
--- a/habitat/core/vector_env.py
+++ b/habitat/core/vector_env.py
@@ -409,7 +409,7 @@ class VectorEnv:
"""Render observations from all environments in a tiled image.
"""
for write_fn in self._connection_write_fns:
- write_fn((args, {"mode": "rgb_array", **kwargs}))
+ write_fn((RENDER_COMMAND, (args, {"mode": "rgb", **kwargs})))
images = [read_fn() for read_fn in self._connection_read_fns]
tile = tile_images(images)
if mode == "human":
diff --git a/test/test_habitat_env.py b/test/test_habitat_env.py
index 3229c9d98..8d10f5855 100644
--- a/test/test_habitat_env.py
+++ b/test/test_habitat_env.py
@@ -221,6 +221,7 @@ def test_rl_vectorized_envs():
assert len(rewards) == num_envs
assert len(dones) == num_envs
assert len(infos) == num_envs
+ assert envs.render(mode="rgb_array"), "vector env render is broken"
if (i + 1) % configs[0].ENVIRONMENT.MAX_EPISODE_STEPS == 0:
assert all(dones), "dones should be true after max_episode steps"
--
GitLab