':param sensor_idx: list or ndarray of indices to be shown. Other indices will be occluded. Can be either list of
integer indices or boolean mask.'

def __init__(self, env, sensor_idx):

    Serializable.quick_init(self, locals())
    self._set_sensor_mask(env, sensor_idx)
    super(OcclusionEnv, self).__init__(env)
    self._dt = 1
    if isinstance(env, MujocoEnv):
        self._dt = (env.model.opt.timestep * env.frame_skip)
