gym.Env actually does not want to known about render_mode

This commit is contained in:
Dominik Moritz Roth 2023-10-23 13:12:34 +02:00
parent f7a493d8e5
commit d7ea1f80a0

View File

@ -14,8 +14,8 @@ class BaseReacherEnv(gym.Env):
Base class for all reaching environments. Base class for all reaching environments.
""" """
def __init__(self, n_links: int, random_start: bool = True, allow_self_collision: bool = False, render_mode: str = None, **kwargs): def __init__(self, n_links: int, random_start: bool = True, allow_self_collision: bool = False, render_mode: str = None):
super().__init__(render_mode=render_mode, **kwargs) super().__init__()
self.link_lengths = np.ones(n_links) self.link_lengths = np.ones(n_links)
self.n_links = n_links self.n_links = n_links
self._dt = 0.01 self._dt = 0.01