dodox/fancy_gym
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

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

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
fancy_gym/envs/classic_control/hole_reacher/hole_reacher.py
View File

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

4
fancy_gym/envs/classic_control/simple_reacher/simple_reacher.py
View File

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

4
fancy_gym/envs/classic_control/viapoint_reacher/viapoint_reacher.py
View File

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