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

Adding required render_fps metadata to custom envs

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-06-11 12:30:56 +02:00
parent ef64b0c21c
commit 6f1a36d18c
4 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
fancy_gym/dmc/dmc_wrapper.py
View File

@ -62,6 +62,8 @@ class DMCWrapper(gym.Env):
env: Callable[[], Union[composer.Environment, control.Environment]], env: Callable[[], Union[composer.Environment, control.Environment]],
): ):
raise Exception('The fancy_gym dmc-wrapper is deprecated; shimmy should be used instead.')
# TODO: Currently this is required to be a function because dmc does not allow to copy composers environments # TODO: Currently this is required to be a function because dmc does not allow to copy composers environments
self._env = env() self._env = env()

1
fancy_gym/envs/mujoco/beerpong/beerpong.py
View File

@ -38,6 +38,7 @@ class BeerPongEnv(MujocoEnv, utils.EzPickle):
"rgb_array", "rgb_array",
"depth_array", "depth_array",
], ],
"render_fps": 100
} }
def __init__(self, **kwargs): def __init__(self, **kwargs):

1
fancy_gym/envs/mujoco/box_pushing/box_pushing_env.py
View File

@ -33,6 +33,7 @@ class BoxPushingEnvBase(MujocoEnv, utils.EzPickle):
"rgb_array", "rgb_array",
"depth_array", "depth_array",
], ],
"render_fps": 50
} }
def __init__(self, frame_skip: int = 10): def __init__(self, frame_skip: int = 10):

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

@ -29,6 +29,7 @@ class TableTennisEnv(MujocoEnv, utils.EzPickle):
"rgb_array", "rgb_array",
"depth_array", "depth_array",
], ],
"render_fps": 125
} }
def __init__(self, ctxt_dim: int = 4, frame_skip: int = 4, def __init__(self, ctxt_dim: int = 4, frame_skip: int = 4,