From 3806d1fcff5e2c67f7ae73213114ffcbdde8bd68 Mon Sep 17 00:00:00 2001
From: Vincent Moens <vmoens@meta.com>
Date: Mon, 24 Feb 2025 16:57:44 +0000
Subject: [PATCH] [BugFix] Fix forced batch-size in _skip_tensordict

ghstack-source-id: dac84e8b8835e870bce1772d7893c30b6f9af59c
Pull Request resolved: https://github.com/pytorch/rl/pull/2808
---
 torchrl/envs/common.py | 1 -
 1 file changed, 1 deletion(-)

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,