From 970fb0eb07bcad3213209f62b5d635289f19d01f Mon Sep 17 00:00:00 2001
From: Vincent Moens <vmoens@meta.com>
Date: Mon, 24 Feb 2025 12:08:30 +0000
Subject: [PATCH] Update (base update)

[ghstack-poisoned]
---
 torchrl/envs/common.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/torchrl/envs/common.py b/torchrl/envs/common.py
index a6f35ce16bc..c87b1856430 100644
--- a/torchrl/envs/common.py
+++ b/torchrl/envs/common.py
@@ -1967,6 +1967,7 @@ 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,