update to align with 265b0ec44d42ea8cb912534df5b1e818d6caec0d

config.yaml

@@ -61,13 +61,16 @@ fps: 10
 env:
   name: pusht
   task: PushT-v0
-  from_pixels: true
-  pixels_only: false
   image_size: 96
-  episode_length: 300
-  fps: ${fps}
   state_dim: 2
   action_dim: 2
+  fps: ${fps}
+  episode_length: 300
+  gym:
+    obs_type: pixels_agent_pos
+    render_mode: rgb_array
+    visualization_width: 384
+    visualization_height: 384
 override_dataset_stats:
   observation.image:
     mean: