Only sets qpa offscreen if rgb_array

f1tenth · Oct 8, 2024 · 34a1d73 · 34a1d73
1 parent c23ad6d
commit 34a1d73
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/f1tenth_gym/envs/rendering/__init__.py b/f1tenth_gym/envs/rendering/__init__.py
@@ -35,7 +35,8 @@ def make_renderer(
     if render_spec.render_type == "pygame":
         from .rendering_pygame import PygameEnvRenderer as EnvRenderer
     elif render_spec.render_type == "pyqt6":
-        os.environ["QT_QPA_PLATFORM"] = "offscreen"
+        if render_mode == "rgb_array":
+            os.environ["QT_QPA_PLATFORM"] = "offscreen"
         from .rendering_pyqt import PyQtEnvRenderer as EnvRenderer
     else:
         raise ValueError(f"Unknown render type: {render_spec.render_type}")