From 6f1a36d18c2149e0ce138f7b2788332d9a888c54 Mon Sep 17 00:00:00 2001
From: Dominik Roth <dominik.roth.dev@gmail.com>
Date: Sun, 11 Jun 2023 12:30:56 +0200
Subject: [PATCH] Adding required render_fps metadata to custom envs

---
 fancy_gym/dmc/dmc_wrapper.py                           | 2 ++
 fancy_gym/envs/mujoco/beerpong/beerpong.py             | 1 +
 fancy_gym/envs/mujoco/box_pushing/box_pushing_env.py   | 1 +
 fancy_gym/envs/mujoco/table_tennis/table_tennis_env.py | 1 +
 4 files changed, 5 insertions(+)

diff --git a/fancy_gym/dmc/dmc_wrapper.py b/fancy_gym/dmc/dmc_wrapper.py
index d1e5f0d..b4f2292 100644
--- a/fancy_gym/dmc/dmc_wrapper.py
+++ b/fancy_gym/dmc/dmc_wrapper.py
@@ -62,6 +62,8 @@ class DMCWrapper(gym.Env):
                  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
         self._env = env()
 
diff --git a/fancy_gym/envs/mujoco/beerpong/beerpong.py b/fancy_gym/envs/mujoco/beerpong/beerpong.py
index 1f35bce..8e2f9fc 100644
--- a/fancy_gym/envs/mujoco/beerpong/beerpong.py
+++ b/fancy_gym/envs/mujoco/beerpong/beerpong.py
@@ -38,6 +38,7 @@ class BeerPongEnv(MujocoEnv, utils.EzPickle):
             "rgb_array",
             "depth_array",
         ],
+        "render_fps": 100
     }
 
     def __init__(self, **kwargs):
diff --git a/fancy_gym/envs/mujoco/box_pushing/box_pushing_env.py b/fancy_gym/envs/mujoco/box_pushing/box_pushing_env.py
index 65db553..3efcf3f 100644
--- a/fancy_gym/envs/mujoco/box_pushing/box_pushing_env.py
+++ b/fancy_gym/envs/mujoco/box_pushing/box_pushing_env.py
@@ -33,6 +33,7 @@ class BoxPushingEnvBase(MujocoEnv, utils.EzPickle):
             "rgb_array",
             "depth_array",
         ],
+        "render_fps": 50
     }
 
     def __init__(self, frame_skip: int = 10):
diff --git a/fancy_gym/envs/mujoco/table_tennis/table_tennis_env.py b/fancy_gym/envs/mujoco/table_tennis/table_tennis_env.py
index a5d67c0..ddf5022 100644
--- a/fancy_gym/envs/mujoco/table_tennis/table_tennis_env.py
+++ b/fancy_gym/envs/mujoco/table_tennis/table_tennis_env.py
@@ -29,6 +29,7 @@ class TableTennisEnv(MujocoEnv, utils.EzPickle):
             "rgb_array",
             "depth_array",
         ],
+        "render_fps": 125
     }
 
     def __init__(self, ctxt_dim: int = 4, frame_skip: int = 4,