[BugFix] Propagate args to TransformedEnv's state_dict (#944)

Author: fedebotu

torchrl/envs/transforms/transforms.py

@@ -523,8 +523,8 @@ def _reset(self, tensordict: Optional[TensorDictBase] = None, **kwargs):
         out_tensordict = self.transform._call(out_tensordict)
         return out_tensordict
 
-    def state_dict(self) -> OrderedDict:
-        state_dict = self.transform.state_dict()
+    def state_dict(self, *args, **kwargs) -> OrderedDict:
+        state_dict = self.transform.state_dict(*args, **kwargs)
         return state_dict
 
     def load_state_dict(self, state_dict: OrderedDict, **kwargs) -> None: