Whalswp commited on
Commit
c827431
·
verified ·
1 Parent(s): b0bb325

Update env.py

Browse files
Files changed (1) hide show
  1. env.py +3 -1
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: