AutoResetEnv

class torchrl.envs.transforms.AutoResetEnv(*args, **kwargs)

A subclass for auto-resetting envs.

insert_transform(index: int, transform: Transform) → None

Inserts a transform to the env at the desired index.

Transform or callable are accepted.