Update env.py
Browse files
env.py
CHANGED
|
@@ -68,9 +68,11 @@ class RoboCasaEnv(RoboCasaGymEnv):
|
|
| 68 |
):
|
| 69 |
self.obs_type = obs_type
|
| 70 |
self.render_mode = render_mode
|
| 71 |
-
self.kwargs = kwargs
|
| 72 |
self.split = split
|
| 73 |
self.task = task
|
|
|
|
|
|
|
|
|
|
| 74 |
|
| 75 |
meta_info = {**ATOMIC_TASK_DATASETS, **COMPOSITE_TASK_DATASETS}
|
| 76 |
try:
|
|
|
|
| 68 |
):
|
| 69 |
self.obs_type = obs_type
|
| 70 |
self.render_mode = render_mode
|
|
|
|
| 71 |
self.split = split
|
| 72 |
self.task = task
|
| 73 |
+
|
| 74 |
+
kwargs.pop("fps", None)
|
| 75 |
+
self.kwargs = kwargs
|
| 76 |
|
| 77 |
meta_info = {**ATOMIC_TASK_DATASETS, **COMPOSITE_TASK_DATASETS}
|
| 78 |
try:
|