diff --git a/torchrl/envs/transforms/transforms.py b/torchrl/envs/transforms/transforms.py index c8e7be93be9..255005ed30e 100644 --- a/torchrl/envs/transforms/transforms.py +++ b/torchrl/envs/transforms/transforms.py @@ -308,7 +308,7 @@ def _reset( self, tensordict: TensorDictBase, tensordict_reset: TensorDictBase ) -> TensorDictBase: """Resets a transform if it is stateful.""" - return tensordict_reset + return self._call(tensordict_reset) def _reset_env_preprocess(self, tensordict: TensorDictBase) -> TensorDictBase: """Inverts the input to :meth:`TransformedEnv._reset`, if needed."""