Update __init__.py

f1tenth · Oct 11, 2024 · b24ad7e · b24ad7e
1 parent df4187e
commit b24ad7e
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 render_mode == "rgb_array":
+        if "rgb_array" in render_mode:
             os.environ["QT_QPA_PLATFORM"] = "offscreen"
         from .rendering_pyqt import PyQtEnvRenderer as EnvRenderer
     else: