File size: 268 Bytes
987ed1b
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
"""Environments using kitchen and Franka robot."""
from gym.envs.registration import register


register(
    id="particle-spread-v0",
    entry_point="diffusion_policy.env.particle.spread_env:ParticleSpreadEnv",
    max_episode_steps=200,
    reward_threshold=1.0,
)