diff --git a/f1tenth_gym/envs/rendering/__init__.py b/f1tenth_gym/envs/rendering/__init__.py index bcfa6bfb..0e52410f 100644 --- 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 index 900a5a3f..ecd2469e 100644 --- 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: