Update (base update)

[ghstack-poisoned]

Author: vmoens

torchrl/envs/batched_envs.py

@@ -1875,7 +1875,6 @@ def _step(self, tensordict: TensorDictBase) -> TensorDictBase:
                 tensordict = tensordict[partial_steps].to(
                     self.shared_tensordict_parent.device
                 )
-                shared_tensordict_parent = shared_tensordict_parent[partial_steps]
         else:
             workers_range = range(self.num_workers)
             shared_tensordict_parent = self.shared_tensordict_parent

torchrl/envs/transforms/transforms.py

@@ -9103,7 +9103,6 @@ class _CallableTransform(Transform):
     # A wrapper around a custom callable to make it possible to transform any data type
     def __init__(self, func):
         super().__init__()
-        raise RuntimeError(isinstance(func, Transform), func)
         self.func = func
 
     def forward(self, *args, **kwargs):