remove 'mode' from render() for all classic_control envs

This commit is contained in:
Dominik Moritz Roth 2023-11-21 20:17:19 +01:00
parent dc773c6c10
commit 01d2bf44ba
3 changed files with 3 additions and 9 deletions

View File

@ -178,9 +178,7 @@ class HoleReacherEnv(BaseReacherDirectEnv):
return False
def render(self, mode=None):
if mode==None:
mode = self.render_mode
def render(self):
if self.fig is None:
# Create base figure once on the beginning. Afterwards only update
plt.ion()

View File

@ -98,9 +98,7 @@ class SimpleReacherEnv(BaseReacherTorqueEnv):
def _check_collisions(self) -> bool:
return self._check_self_collision()
def render(self, mode=None): # pragma: no cover
if mode==None:
mode = self.render_mode
def render(self): # pragma: no cover
if self.fig is None:
# Create base figure once on the beginning. Afterwards only update
plt.ion()

View File

@ -123,9 +123,7 @@ class ViaPointReacherEnv(BaseReacherDirectEnv):
def _check_collisions(self) -> bool:
return self._check_self_collision()
def render(self, mode=None):
if mode==None:
mode = self.render_mode
def render(self):
goal_pos = self._goal.T
via_pos = self._via_point.T