Method _reset

advanced_source/pendulum.py:680–683

(
        self, tensordict: TensorDictBase, tensordict_reset: TensorDictBase
    )

    # The transform must also modify the data at reset time
    def _reset(
        self, tensordict: TensorDictBase, tensordict_reset: TensorDictBase
    ) -> TensorDictBase:
        return self._call(tensordict_reset)

    # _apply_to_composite will execute the observation spec transform across all
    # in_keys/out_keys pairs and write the result in the observation_spec which