diff --git a/torchrl/envs/common.py b/torchrl/envs/common.py index c87b1856430..a6f35ce16bc 100644 --- a/torchrl/envs/common.py +++ b/torchrl/envs/common.py @@ -1967,7 +1967,6 @@ def select_and_clone(name, x, y): select_and_clone, next_tensordict, device=next_tensordict.device, - batch_size=next_tensordict.batch_size, default=None, filter_empty=True, is_leaf=_is_leaf_nontensor,