diff --git a/habitat-lab/habitat/core/batch_rendering/env_batch_renderer.py b/habitat-lab/habitat/core/batch_rendering/env_batch_renderer.py index 5480fd1ec2..8ee92821af 100644 --- a/habitat-lab/habitat/core/batch_rendering/env_batch_renderer.py +++ b/habitat-lab/habitat/core/batch_rendering/env_batch_renderer.py @@ -224,7 +224,7 @@ def _draw_observations_gpu_to_cpu( raise NotImplementedError # Render - self._replay_renderer.render(self._gpu_to_cpu_images) + self._replay_renderer.render(color_images=self._gpu_to_cpu_images) return self._gpu_to_cpu_buffer def _draw_observations_gpu_to_gpu(