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

BugFix: DId not pass kwargs down in new TT envs

Browse Source
This commit is contained in:
Dominik Moritz Roth 2024-01-28 12:54:49 +01:00
parent db8221ebb2
commit 3bc0a23ec2
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
fancy_gym/envs/mujoco/__init__.py
View File

@ -8,7 +8,7 @@ from .hopper_throw.hopper_throw_in_basket import HopperThrowInBasketEnv
from .reacher.reacher import ReacherEnv from .reacher.reacher import ReacherEnv
from .walker_2d_jump.walker_2d_jump import Walker2dJumpEnv from .walker_2d_jump.walker_2d_jump import Walker2dJumpEnv
from .box_pushing.box_pushing_env import BoxPushingDense, BoxPushingTemporalSparse, BoxPushingTemporalSpatialSparse from .box_pushing.box_pushing_env import BoxPushingDense, BoxPushingTemporalSparse, BoxPushingTemporalSpatialSparse
from .table_tennis.table_tennis_env import TableTennisEnv, TableTennisWind, TableTennisGoalSwitching from .table_tennis.table_tennis_env import TableTennisEnv, TableTennisWind, TableTennisGoalSwitching, TableTennisMarkovian, TableTennisRandomInit
try: try:
from .air_hockey.air_hockey_env_wrapper import AirHockeyEnv from .air_hockey.air_hockey_env_wrapper import AirHockeyEnv

8
fancy_gym/envs/mujoco/table_tennis/table_tennis_env.py
View File

@ -41,6 +41,7 @@ class TableTennisEnv(MujocoEnv, utils.EzPickle):
enable_artificial_wind: bool = False, enable_artificial_wind: bool = False,
random_pos_scale: float = 0.0, random_pos_scale: float = 0.0,
random_vel_scale: float = 0.0, random_vel_scale: float = 0.0,
**kwargs,
): ):
utils.EzPickle.__init__(**locals()) utils.EzPickle.__init__(**locals())
self._steps = 0 self._steps = 0
@ -490,8 +491,9 @@ class TableTennisGoalSwitching(TableTennisEnv):
class TableTennisRandomInit(TableTennisEnv): class TableTennisRandomInit(TableTennisEnv):
def __init__(self, ctxt_dim: int = 4, frame_skip: int = 4, def __init__(self, ctxt_dim: int = 4, frame_skip: int = 4,
random_pos_scale: float = 1.0, random_pos_scale: float = 1.0,
random_vel_scale: float = 0.0): random_vel_scale: float = 0.0,
**kwargs):
super().__init__(ctxt_dim=ctxt_dim, frame_skip=frame_skip, super().__init__(ctxt_dim=ctxt_dim, frame_skip=frame_skip,
random_pos_scale=random_pos_scale, random_pos_scale=random_pos_scale,
random_vel_scale=random_vel_scale) random_vel_scale=random_vel_scale,
**kwargs)