From d5310db52b6732da3f0c017fd9bd484555764d8a Mon Sep 17 00:00:00 2001 From: Louis Faury Date: Tue, 22 Apr 2025 22:43:48 +0200 Subject: [PATCH] Base transform applies on reset --- torchrl/envs/transforms/transforms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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."""