Update __init__.py

f1tenth · Oct 11, 2024 · 5ba30b4 · 5ba30b4
1 parent b24ad7e
commit 5ba30b4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/f1tenth_gym/envs/rendering/__init__.py b/f1tenth_gym/envs/rendering/__init__.py
@@ -35,7 +35,7 @@ def make_renderer(
     if render_spec.render_type == "pygame":
         from .rendering_pygame import PygameEnvRenderer as EnvRenderer
     elif render_spec.render_type == "pyqt6":
-        if "rgb_array" in render_mode:
+        if render_mode in ["rgb_array", "rgb_array_list"]:
             os.environ["QT_QPA_PLATFORM"] = "offscreen"
         from .rendering_pyqt import PyQtEnvRenderer as EnvRenderer
     else: