Merge pull request StanfordVL#54 from LinxiFan/master

minor fix
dnandha · Jun 3, 2020 · 48cc526 · 48cc526
2 parents a3b1219 + 509e257
commit 48cc526
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gibson2/core/render/mesh_renderer/mesh_renderer_cpu.py b/gibson2/core/render/mesh_renderer/mesh_renderer_cpu.py
@@ -325,7 +325,7 @@ def __init__(self, width=512, height=512, vertical_fov=90, device_idx=0, use_fis
         self.fisheye = use_fisheye
         # self.context = glcontext.Context()
         # self.context.create_opengl_context((self.width, self.height))
-        if 'GIBSON_DEVICE_ID' in os.environ:
+        if os.environ.get('GIBSON_DEVICE_ID', None):
             device = int(os.environ.get('GIBSON_DEVICE_ID'))
             logging.info(f'GIBSON_DEVICE_ID environment variable has been manually set. '
                          f'Using device {device} for rendering')