Fixing last commit

f1tenth · Oct 2, 2024 · 967e1aa · 967e1aa
1 parent 67d72cf
commit 967e1aa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/f1tenth_gym/envs/rendering/__init__.py b/f1tenth_gym/envs/rendering/__init__.py
@@ -1,4 +1,5 @@
 import pathlib
+import os
 from typing import Any, Optional
 
 from .renderer import RenderSpec, EnvRenderer
@@ -34,6 +35,7 @@ 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"
         from .rendering_pyqt import PyQtEnvRenderer as EnvRenderer
     else:
         raise ValueError(f"Unknown render type: {render_spec.render_type}")

diff --git a/f1tenth_gym/envs/rendering/rendering_pyqt.py b/f1tenth_gym/envs/rendering/rendering_pyqt.py
@@ -3,7 +3,6 @@
 import math
 from typing import Any, Callable, Optional
 import signal
-import os
 
 import cv2
 import numpy as np
@@ -165,7 +164,6 @@ def __init__(
             signal.signal(signal.SIGINT, signal.SIG_DFL)
             self.window.show()
         elif self.render_mode == "rgb_array":
-            os.environ["QT_QPA_PLATFORM"] = "offscreen"
             self.exporter = ImageExporter(self.canvas)
 
     def update(self, state: dict) -> None: