File size: 238 Bytes
987ed1b | 1 2 3 4 5 6 7 8 9 | from gym.envs.registration import register
import diffusion_policy.env.pusht
register(
id='pusht-keypoints-v0',
entry_point='envs.pusht.pusht_keypoints_env:PushTKeypointsEnv',
max_episode_steps=200,
reward_threshold=1.0
) |